treelite
Classes | Enumerations | Functions
cli_main.cc File Reference

The command line interface program of treelite. More...

#include <treelite/frontend.h>
#include <treelite/annotator.h>
#include <treelite/compiler.h>
#include <treelite/semantic.h>
#include <treelite/predictor.h>
#include <treelite/logging.h>
#include <treelite/omp.h>
#include <dmlc/config.h>
#include <dmlc/data.h>
#include <fstream>
#include <memory>
#include <vector>
#include <queue>
#include <iterator>
#include <string>
#include "./compiler/param.h"
#include "./common/filesystem.h"
Include dependency graph for cli_main.cc:

Go to the source code of this file.

Classes

struct  CLIParam
 

Enumerations

enum  CLITask { kCodegen = 0, kAnnotate = 1, kPredict = 2 }
 
enum  InputFormat { kLibSVM = 0, kCSV = 1, kLibFM = 2 }
 
enum  ModelFormat { kXGBModel = 0, kLGBModel = 1, kProtobuf = 2 }
 

Functions

const char * FileFormatString (int format)
 
 DMLC_REGISTER_PARAMETER (CLIParam)
 
Model ParseModel (const CLIParam &param)
 
void CLICodegen (const CLIParam &param)
 
void CLIAnnotate (const CLIParam &param)
 
void CLIPredict (const CLIParam &param)
 
int CLIRunTask (int argc, char *argv[])
 
int main (int argc, char *argv[])
 

Detailed Description

The command line interface program of treelite.

Copyright 2017 by Contributors

Author
Philip Cho

Definition in file cli_main.cc.