rmontanana/mdlp
1
1
Fork 0
mirror of https://github.com/rmontanana/mdlp.git synced 2025-08-16 07:55:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add max_depth and min_length as hyperparams

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-02-25 18:16:20 +01:00
parent e25ca378f0
commit d6cece1006
4 changed files with 105 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
CPPFImdlp.h
View File

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