MedBART.h

#ifndef _MED_BART_H_
#define _MED_BART_H_
#include "MedAlgo.h"
#include "BART.h"
 
class MedBART : public MedPredictor {
public:
 
    void init_defaults() {};
    int set_params(map<string, string>& mapper);
 
    int Learn(float *x, float *y, const float *w, int nsamples, int nftrs);
    int Predict(float *x, float *&preds, int nsamples, int nftrs) const;
 
    MedBART() : _model(0, 0, 0, 0, tree_params) {
        normalize_for_learn = false;
        transpose_for_learn = false;
        normalize_y_for_learn = false;
        transpose_for_predict = false;
        normalize_for_predict = false;
        classifier_type = MODEL_BART;
 
        //default params:
        ntrees = 50;
        iter_count = 1000;
        burn_count = 250;
        restart_count = 5;
        tree_params.alpha = (float)0.95;
        tree_params.beta = 1;
        tree_params.min_obs_in_node = 5;
 
        tree_params.k = 2;
        tree_params.nu = 3;
        tree_params.lambda = 1;
    }
 
    ADD_CLASS_NAME(MedBART)
    ADD_SERIALIZATION_FUNCS(classifier_type, ntrees, iter_count, burn_count, restart_count)
private:
    int ntrees; 
    int iter_count; 
    int burn_count; 
    int restart_count; 
    bart_params tree_params; 
 
    BART _model;
};
 
MEDSERIALIZE_SUPPORT(MedBART)
 
#endif // !_MED_BART_H_