mirror of
https://github.com/rmontanana/mdlp.git
synced 2025-08-20 18:06:00 +00:00
e36d9af8f9
* Fix BinDisc quantile mistakes * Fix FImdlp tests * Fix tests, samples and remove uneeded support files * Add coypright header to sources Fix coverage report Add coverage badge to README * Update sonar github action * Move sources to a folder and change ArffFiles files to library * Add recursive submodules to github action
28 lines
828 B
C++
28 lines
828 B
C++
// ****************************************************************
|
|
// SPDX - FileCopyrightText: Copyright 2024 Ricardo Montañana Gómez
|
|
// SPDX - FileType: SOURCE
|
|
// SPDX - License - Identifier: MIT
|
|
// ****************************************************************
|
|
|
|
#ifndef CCMETRICS_H
|
|
#define CCMETRICS_H
|
|
|
|
#include "typesFImdlp.h"
|
|
|
|
namespace mdlp {
|
|
class Metrics {
|
|
protected:
|
|
labels_t& y;
|
|
indices_t& indices;
|
|
int numClasses;
|
|
cacheEnt_t entropyCache = cacheEnt_t();
|
|
cacheIg_t igCache = cacheIg_t();
|
|
public:
|
|
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);
|
|
};
|
|
}
|
|
#endif |