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 <limits>
#include <queue>
Include dependency graph for lightgbm.cc:

Go to the source code of this file.

Enumerations

enum  Masks : uint8_t
 
enum  MissingType : uint8_t
 

Functions

 DMLC_REGISTRY_FILE_TAG (lightgbm)
 
void LoadLightGBMModel (const char *filename, Model *out)
 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 (c) 2017-2020 by Contributors

Author
Hyunsu Cho

Definition in file lightgbm.cc.

Function Documentation

◆ LoadLightGBMModel()

void LoadLightGBMModel ( const char *  filename,
Model out 
)

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

Parameters
filenamename of model file
outreference to loaded model

Definition at line 26 of file lightgbm.cc.