Treelite
Classes | Enumerations | Functions | Variables
xgboost.h File Reference

Helper functions for loading XGBoost models. More...

#include <string>
#include <vector>
#include <cmath>
Include dependency graph for xgboost.h:
This graph shows which files directly or indirectly include this file:

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
 

Detailed Description

Helper functions for loading XGBoost models.

Copyright (c) 2020 by Contributors

Author
William Hicks

Definition in file xgboost.h.

Variable Documentation

◆ exponential_objectives

const std::vector< std::string > exponential_objectives
Initial value:
{
"count:poisson", "reg:gamma", "reg:tweedie", "survival:cox", "survival:aft"
}

Definition at line 25 of file xgboost_util.cc.