Treelite
Classes | Macros | Functions
tree.h File Reference

model structure for tree ensemble More...

#include <treelite/base.h>
#include <algorithm>
#include <map>
#include <string>
#include <vector>
#include <utility>
#include <type_traits>
#include <limits>
#include <cstring>
#include <cstdio>
#include "tree_impl.h"
Include dependency graph for tree.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  PyBufferFrame
 
class  ContiguousArray< T >
 
class  Tree
 in-memory representation of a decision tree More...
 
struct  Tree::Node
 tree node More...
 
union  Tree::Node::Info
 store either leaf value or decision threshold More...
 
struct  ModelParam
 
struct  Model
 thin wrapper for tree ensemble model More...
 

Macros

#define __TREELITE_STR(x)   #x
 
#define _TREELITE_STR(x)   __TREELITE_STR(x)
 
#define TREELITE_MAX_PRED_TRANSFORM_LENGTH   256
 

Functions

float stof (const std::string &value, size_t *pos)
 
void InitParamAndCheck (ModelParam *param, const std::vector< std::pair< std::string, std::string >> &cfg)
 

Detailed Description

model structure for tree ensemble

Copyright (c) 2017-2020 by Contributors

Author
Hyunsu Cho

Definition in file tree.h.