Fix entropy and information gain

This commit is contained in:
2022-11-28 11:39:11 +01:00
parent 3d27c4c3b7
commit e9e2a66203
4 changed files with 53 additions and 22 deletions

View File

@@ -9,8 +9,9 @@ namespace CPPFImdlp
{
public:
Metrics();
static float entropy(std::vector<int> &, int, int, int);
static int numClasses(std::vector<int> &);
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);
};
}
#endif