mirror of
https://github.com/rmontanana/mdlp.git
synced 2025-08-16 07:55:58 +00:00
* Add better check in testKBins.py * Add Discretizer base class for Both discretizers * Refactor order of constructors init
39 lines
1.2 KiB
C++
39 lines
1.2 KiB
C++
#ifndef CPPFIMDLP_H
|
|
#define CPPFIMDLP_H
|
|
|
|
#include "typesFImdlp.h"
|
|
#include <limits>
|
|
#include <utility>
|
|
#include <string>
|
|
#include "Metrics.h"
|
|
#include "Discretizer.h"
|
|
|
|
namespace mdlp {
|
|
class CPPFImdlp : public Discretizer {
|
|
public:
|
|
CPPFImdlp() = default;
|
|
CPPFImdlp(size_t min_length_, int max_depth_, float proposed);
|
|
virtual ~CPPFImdlp() = default;
|
|
void fit(samples_t& X_, labels_t& y_) override;
|
|
inline int get_depth() const { return depth; };
|
|
protected:
|
|
size_t min_length = 3;
|
|
int depth = 0;
|
|
int max_depth = numeric_limits<int>::max();
|
|
float proposed_cuts = 0;
|
|
indices_t indices = indices_t();
|
|
samples_t X = samples_t();
|
|
labels_t y = labels_t();
|
|
Metrics metrics = Metrics(y, indices);
|
|
size_t num_cut_points = numeric_limits<size_t>::max();
|
|
static indices_t sortIndices(samples_t&, labels_t&);
|
|
void computeCutPoints(size_t, size_t, int);
|
|
void resizeCutPoints();
|
|
bool mdlp(size_t, size_t, size_t);
|
|
size_t getCandidate(size_t, size_t);
|
|
size_t compute_max_num_cut_points() const;
|
|
pair<precision_t, size_t> valueCutPoint(size_t, size_t, size_t);
|
|
};
|
|
}
|
|
#endif
|