Remove unneeded loop in sortIndices

Add some static casts
This commit is contained in:
2023-03-19 19:13:37 +01:00
parent f0845c5bd1
commit cfade7a556
9 changed files with 210 additions and 146 deletions

View File

@@ -1,19 +1,25 @@
#ifndef CCMETRICS_H
#define CCMETRICS_H
#include "typesFImdlp.h"
namespace mdlp {
class Metrics {
protected:
labels_t& y;
indices_t& indices;
labels_t &y;
indices_t &indices;
int numClasses;
cacheEnt_t entropyCache;
cacheIg_t igCache;
cacheEnt_t entropyCache = cacheEnt_t();
cacheIg_t igCache = cacheIg_t();
public:
Metrics(labels_t&, indices_t&);
void setData(labels_t&, indices_t&);
Metrics(labels_t &, indices_t &);
void setData(const labels_t &, const indices_t &);
int computeNumClasses(size_t, size_t);
precision_t entropy(size_t, size_t);
precision_t informationGain(size_t, size_t, size_t);
};
}