Doctorado-ML/fimdlp
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/FImdlp.git synced 2025-08-18 17:05:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix entroy and ig

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2022-11-28 18:03:46 +01:00
parent e9e2a66203
commit 6c507a24b0
7 changed files with 34 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
fimdlp/Metrics.h
View File

@@ -9,9 +9,9 @@ namespace CPPFImdlp
{
public:
Metrics();
static int numClasses(std::vector<int> &, std::vector<size_t>, int, int);
static float entropy(std::vector<int> &, std::vector<size_t> &, int, int, int);
static float informationGain(std::vector<int> &y, std::vector<size_t> &indices, int start, int end, int cutPoint, int nClasses);
static int numClasses(std::vector<int> &, std::vector<size_t>, size_t, size_t);
static float entropy(std::vector<int> &, std::vector<size_t> &, size_t, size_t, int);
static float informationGain(std::vector<int> &, std::vector<size_t> &, size_t, size_t, size_t, int);
};
}
#endif