Treelite
|
Helper functions for loading XGBoost models. More...
#include <string>
#include <vector>
#include <cmath>
Go to the source code of this file.
Classes | |
struct | ProbToMargin |
Enumerations | |
enum | FeatureType { kNumerical = 0, kCategorical = 1 } |
Functions | |
void | SetPredTransform (const std::string &objective_name, ModelParam *param) |
void | TransformGlobalBiasToMargin (ModelParam *param) |
Variables | |
const std::vector< std::string > | exponential_objectives |
Helper functions for loading XGBoost models.
Copyright (c) 2020 by Contributors
Definition in file xgboost.h.
const std::vector< std::string > exponential_objectives |
Definition at line 25 of file xgboost_util.cc.