treelite
Enumerations | Functions
lightgbm.cc File Reference

Frontend for lightgbm model. More...

#include <dmlc/data.h>
#include <treelite/frontend.h>
#include <treelite/tree.h>
#include <unordered_map>
#include <queue>
Include dependency graph for lightgbm.cc:

Go to the source code of this file.

Enumerations

enum  Masks : uint8_t
 

Functions

 DMLC_REGISTRY_FILE_TAG (lightgbm)
 
Model LoadLightGBMModel (const char *filename)
 load a model file generated by LightGBM (Microsoft/LightGBM). The model file must contain a decision tree ensemble. More...
 

Detailed Description

Frontend for lightgbm model.

Copyright 2017 by Contributors

Author
Philip Cho

Definition in file lightgbm.cc.

Function Documentation

Model LoadLightGBMModel ( const char *  filename)

load a model file generated by LightGBM (Microsoft/LightGBM). The model file must contain a decision tree ensemble.

Parameters
filenamename of model file
Returns
loaded model

Definition at line 25 of file lightgbm.cc.