mirror of
https://github.com/Doctorado-ML/FImdlp.git
synced 2025-08-17 16:35:52 +00:00
21 lines
514 B
C++
21 lines
514 B
C++
#ifndef CCMETRICS_H
|
|
#define CCMETRICS_H
|
|
#include "typesFImdlp.h"
|
|
#include <cmath>
|
|
namespace mdlp {
|
|
class Metrics {
|
|
protected:
|
|
labels& y;
|
|
indices_t& indices;
|
|
int numClasses;
|
|
cacheEnt_t entropyCache;
|
|
cacheIg_t igCache;
|
|
public:
|
|
Metrics(labels&, indices_t&);
|
|
void setData(labels&, indices_t&);
|
|
int computeNumClasses(size_t, size_t);
|
|
float entropy(size_t, size_t);
|
|
float informationGain(size_t, size_t, size_t);
|
|
};
|
|
}
|
|
#endif |