Treelite
|
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"
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) |
model structure for tree ensemble
Copyright (c) 2017-2020 by Contributors
Definition in file tree.h.