Treelite
Functions
xgboost_util.cc File Reference

Common utilities for XGBoost frontends. More...

#include <treelite/tree.h>
#include <treelite/logging.h>
#include <cstring>
#include "xgboost/xgboost.h"
Include dependency graph for xgboost_util.cc:

Go to the source code of this file.

Functions

void SetPredTransform (const std::string &objective_name, ModelParam *param)
 
void TransformGlobalBiasToMargin (ModelParam *param)
 

Detailed Description

Common utilities for XGBoost frontends.

Copyright (c) 2020-2021 by Contributors

Author
Hyunsu Cho

Definition in file xgboost_util.cc.