Treelite
|
Frontend for xgboost model. More...
#include "xgboost/xgboost_json.h"
#include <dmlc/registry.h>
#include <dmlc/io.h>
#include <fmt/format.h>
#include <rapidjson/error/en.h>
#include <rapidjson/document.h>
#include <rapidjson/filereadstream.h>
#include <treelite/tree.h>
#include <treelite/frontend.h>
#include <treelite/math.h>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <memory>
#include <queue>
#include <string>
#include <utility>
#include "xgboost/xgboost.h"
Go to the source code of this file.
Functions | |
DMLC_REGISTRY_FILE_TAG (xgboost_json) | |
std::unique_ptr< treelite::Model > | LoadXGBoostJSONModel (const char *filename) |
load a model file generated by XGBoost (dmlc/xgboost). The model file must contain a decision tree ensemble in the JSON format. More... | |
std::unique_ptr< treelite::Model > | LoadXGBoostJSONModelString (const char *json_str, size_t length) |
load an XGBoost model from a JSON string More... | |
std::unique_ptr< treelite::Model > LoadXGBoostJSONModel | ( | const char * | filename | ) |
load a model file generated by XGBoost (dmlc/xgboost). The model file must contain a decision tree ensemble in the JSON format.
filename | name of model file |
out | reference to loaded model |
Definition at line 45 of file xgboost_json.cc.
std::unique_ptr< treelite::Model > LoadXGBoostJSONModelString | ( | const char * | json_str, |
size_t | length | ||
) |
load an XGBoost model from a JSON string
json_str | JSON char array |
length | length of JSON char array |
out | reference to loaded model |
Definition at line 85 of file xgboost_json.cc.