Add max_cutpoints Hyperparameter

This commit is contained in:
2023-03-13 01:17:04 +01:00
parent ed784736ca
commit ffb8df4d1c
7 changed files with 105 additions and 41 deletions

View File

@@ -9,22 +9,25 @@ namespace mdlp {
protected:
size_t min_length;
int depth, max_depth;
float proposed_cuts;
indices_t indices;
samples_t X;
labels_t y;
indices_t indices;
Metrics metrics;
cutPoints_t cutPoints;
size_t num_cut_points;
static indices_t sortIndices(samples_t&, labels_t&);
void computeCutPoints(size_t, size_t, int);
bool mdlp(size_t, size_t, size_t);
size_t getCandidate(size_t, size_t);
size_t compute_max_num_cut_points();
pair<precision_t, size_t> valueCutPoint(size_t, size_t, size_t);
public:
CPPFImdlp();
CPPFImdlp(size_t, int);
CPPFImdlp(size_t, int, float);
~CPPFImdlp();
CPPFImdlp& fit(samples_t&, labels_t&);
void fit(samples_t&, labels_t&);
cutPoints_t getCutPoints();
int get_depth();
inline string version() { return "1.1.1"; };