7 #ifndef TREELITE_FRONTEND_XGBOOST_XGBOOST_H_ 8 #define TREELITE_FRONTEND_XGBOOST_XGBOOST_H_ 22 static float Sigmoid(
float global_bias) {
23 return -logf(1.0f / global_bias - 1.0f);
25 static float Exponential(
float global_bias) {
26 return logf(global_bias);
30 extern const std::vector<std::string> exponential_objectives;
33 void SetPredTransform(
const std::string& objective_name,
ModelParam* param);
36 void TransformGlobalBiasToMargin(
ModelParam* param);
46 #endif // TREELITE_FRONTEND_XGBOOST_XGBOOST_H_