prediction_early_stop.h

#ifndef LIGHTGBM_PREDICTION_EARLY_STOP_H_
#define LIGHTGBM_PREDICTION_EARLY_STOP_H_
 
#include <functional>
#include <string>
 
#include <LightGBM/export.h>
 
namespace LightGBM {
 
struct PredictionEarlyStopInstance {
  using FunctionType = std::function<bool(const double*, int)>;
 
  FunctionType callback_function;  // callback function itself
  int          round_period;       // call callback_function every `runPeriod` iterations
};
 
struct PredictionEarlyStopConfig {
  int round_period;
  double margin_threshold;
};
 
LIGHTGBM_EXPORT PredictionEarlyStopInstance CreatePredictionEarlyStopInstance(const std::string& type,
                                                                              const PredictionEarlyStopConfig& config);
 
}   // namespace LightGBM
 
#endif  // LIGHTGBM_PREDICTION_EARLY_STOP_H_