mirror of
https://github.com/Doctorado-ML/FImdlp.git
synced 2025-08-17 16:35:52 +00:00
Fix entropy and information gain
This commit is contained in:
@@ -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
|
Reference in New Issue
Block a user