gbm.hpp

#ifndef GBM_H
#define GBM_H

#include <LightGBM/config.h>
#include <LightGBM/dataset_loader.h>
#include <LightGBM/boosting.h>
#include <LightGBM/objective_function.h>
#include <LightGBM/metric.h>
#include <LightGBM/utils/common.h>
#include <LightGBM/prediction_early_stop.h>

#include "vendor/json/json.hpp"
#include "features.hpp"
#include "labels.hpp"
#include "constants.hpp"
#include "point_io.hpp"

using json = nlohmann::json;

namespace gbm {

typedef LightGBM::Boosting Boosting;

Boosting *train(const std::vector<std::string> &filenames,
    double *startResolution,
    int numScales,
    int numTrees,
    int treeDepth,
    double radius,
    int maxSamples,
    const std::vector<int> &classes
);

struct BoosterParams {
    double resolution;
    double radius;
    int numScales;
};

Boosting *loadBooster(const std::string &modelFilename);
void saveBooster(Boosting *booster, const std::string &modelFilename);

BoosterParams extractBoosterParams(Boosting *booster);

void classify(PointSet &pointSet,
    Boosting *booster,
    const std::vector<Feature *> &features,
    const std::vector<Label> &labels,
    Regularization regularization = Regularization::None,
    double regRadius = 2.5,
    bool useColors = false,
    bool unclassifiedOnly = false,
    bool evaluate = false,
    const std::vector<int> &skip = {},
    const std::string &statsFile = "");

}


#endif