You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
843 B
47 lines
843 B
#ifndef RF_H
|
|
#define RF_H
|
|
|
|
#include "DecisionTree.h"
|
|
|
|
struct minEval;
|
|
|
|
struct DR;
|
|
|
|
struct DT;
|
|
|
|
//enum Evaluation {gini, entropy, logLoss};
|
|
|
|
class RandomForest{
|
|
public:
|
|
|
|
long noTree;
|
|
long maxTree;
|
|
long activeTree;
|
|
long treePointer;
|
|
long rotateTime;
|
|
long timer;
|
|
long retain;
|
|
DecisionTree** DTrees = nullptr;
|
|
|
|
long height;
|
|
long Rebuild;
|
|
long f;
|
|
int* sparse;
|
|
double forget;
|
|
long maxFeature;
|
|
long noClasses;
|
|
Evaluation e;
|
|
|
|
|
|
RandomForest(long maxTree, long activeTree, long rotateTime, int height, long f, int* sparse, double forget, long maxFeature=0, long noClasses=2, Evaluation e=Evaluation::gini, long r=-1, long rb=2147483647);
|
|
|
|
void fit(double** data, long* result, long size);
|
|
|
|
long* fitThenPredict(double** trainData, long* trainResult, long trainSize, double** testData, long testSize);
|
|
|
|
void Rotate();
|
|
|
|
long Test(double* data);
|
|
};
|
|
#endif
|