application.h

#ifndef LIGHTGBM_APPLICATION_H_
#define LIGHTGBM_APPLICATION_H_
 
#include <LightGBM/meta.h>
#include <LightGBM/config.h>
 
#include <vector>
#include <memory>
 
namespace LightGBM {
 
class DatasetLoader;
class Dataset;
class Boosting;
class ObjectiveFunction;
class Metric;
 
class Application {
    friend class MemApp; // ADDED by Medial to enable full control inheritance
public:
  Application(int argc, char** argv);
 
  ~Application();
 
  inline void Run();
 
private:
  void LoadParameters(int argc, char** argv);
 
  void LoadData();
 
  void InitTrain();
 
  void Train();
 
  void InitPredict();
 
  void Predict();
 
  void ConvertModel();
 
  Config config_;
  std::unique_ptr<Dataset> train_data_;
  std::vector<std::unique_ptr<Dataset>> valid_datas_;
  std::vector<std::unique_ptr<Metric>> train_metric_;
  std::vector<std::vector<std::unique_ptr<Metric>>> valid_metrics_;
  std::unique_ptr<Boosting> boosting_;
  std::unique_ptr<ObjectiveFunction> objective_fun_;
};
 
 
inline void Application::Run() {
  if (config_.task == TaskType::kPredict || config_.task == TaskType::KRefitTree) {
    InitPredict();
    Predict();
  } else if (config_.task == TaskType::kConvertModel) {
    ConvertModel();
  } else {
    InitTrain();
    Train();
  }
}
 
}  // namespace LightGBM
 
#endif   // LightGBM_APPLICATION_H_