AggregatePredsPostProcessor.h

#ifndef __AVERAGE_PREDS_POST_PROCESSOR_H__
#define __AVERAGE_PREDS_POST_PROCESSOR_H__
 
#include <vector>
#include <Logger/Logger/Logger.h>
#include "PostProcessor.h"
 
using namespace std;
 
class AggregatePredsPostProcessor : public PostProcessor {
private: 
    MedPredictor *model_predictor = NULL; 
    FeatureProcessor *feature_processor = NULL; 
    vector<FeatureProcessor *> before_processors; 
    vector<FeatureProcessor *> after_processors; 
 
    MedModel *p_model; 
 
    void generate_matrix_till_feature_process(const MedFeatures &input_mat, MedFeatures &res) const;
public:
    string feature_processor_type; 
    string feature_processor_args; 
 
    bool force_cancel_imputations; 
    bool use_median; 
    int resample_cnt; 
    int batch_size; 
    bool print_missing_cnt;
    //Add: batch_size, resample_cnt, store_mean (or median)
 
    AggregatePredsPostProcessor();
 
    void get_input_fields(vector<Effected_Field> &fields) const;
    void get_output_fields(vector<Effected_Field> &fields) const;
 
    int init(map<string, string> &mapper);
 
    void Learn(const MedFeatures &train_mat);
    void Apply(MedFeatures &matrix); 
 
    void init_post_processor(MedModel& model);
 
    void dprint(const string &pref) const;
 
    virtual ~AggregatePredsPostProcessor();
 
    ADD_CLASS_NAME(AggregatePredsPostProcessor)
        ADD_SERIALIZATION_FUNCS(model_predictor, feature_processor, feature_processor_type, 
            feature_processor_args, use_median, resample_cnt, batch_size, force_cancel_imputations, 
            before_processors, after_processors, print_missing_cnt)
};
 
MEDSERIALIZE_SUPPORT(AggregatePredsPostProcessor)
 
#endif