Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions
LightGBM::Booster Class Reference

Public Member Functions

 Booster (const char *filename)
 
 Booster (const Dataset *train_data, const char *parameters)
 
void MergeFrom (const Booster *other)
 
void CreateObjectiveAndMetrics ()
 
void ResetTrainingData (const Dataset *train_data)
 
void ResetConfig (const char *parameters)
 
void AddValidData (const Dataset *valid_data)
 
bool TrainOneIter ()
 
void Refit (const int32_t *leaf_preds, int32_t nrow, int32_t ncol)
 
bool TrainOneIter (const score_t *gradients, const score_t *hessians)
 
void RollbackOneIter ()
 
void Predict (int num_iteration, int predict_type, int nrow, std::function< std::vector< std::pair< int, double > >(int row_idx)> get_row_fun, const Config &config, double *out_result, int64_t *out_len)
 
void Predict (int num_iteration, int predict_type, const char *data_filename, int data_has_header, const Config &config, const char *result_filename)
 
void GetPredictAt (int data_idx, double *out_result, int64_t *out_len)
 
void SaveModelToFile (int start_iteration, int num_iteration, const char *filename)
 
void LoadModelFromString (const char *model_str)
 
std::string SaveModelToString (int start_iteration, int num_iteration)
 
std::string DumpModel (int start_iteration, int num_iteration)
 
std::vector< double > FeatureImportance (int num_iteration, int importance_type)
 
double GetLeafValue (int tree_idx, int leaf_idx) const
 
void SetLeafValue (int tree_idx, int leaf_idx, double val)
 
void ShuffleModels (int start_iter, int end_iter)
 
int GetEvalCounts () const
 
int GetEvalNames (char **out_strs) const
 
int GetFeatureNames (char **out_strs) const
 
const BoostingGetBoosting () const
 

The documentation for this class was generated from the following file: