used to find split candidates for a leaf
More...
#include <leaf_splits.hpp>
|
|
| LeafSplits (data_size_t num_data) |
| |
|
void | ResetNumData (data_size_t num_data) |
| |
| void | Init (int leaf, const DataPartition *data_partition, double sum_gradients, double sum_hessians) |
| | Init split on current leaf on partial data.
|
| |
|
void | SetValueConstraint (double min, double max) |
| |
| void | Init (const score_t *gradients, const score_t *hessians) |
| | Init splits on current leaf, it will traverse all data to sum up the results.
|
| |
| void | Init (int leaf, const DataPartition *data_partition, const score_t *gradients, const score_t *hessians) |
| | Init splits on current leaf of partial data.
|
| |
| void | Init (double sum_gradients, double sum_hessians) |
| | Init splits on current leaf, only update sum_gradients and sum_hessians.
|
| |
|
void | Init () |
| | Init splits on current leaf.
|
| |
|
int | LeafIndex () const |
| | Get current leaf index.
|
| |
|
data_size_t | num_data_in_leaf () const |
| | Get numer of data in current leaf.
|
| |
|
double | sum_gradients () const |
| | Get sum of gradients of current leaf.
|
| |
|
double | sum_hessians () const |
| | Get sum of hessians of current leaf.
|
| |
|
double | max_constraint () const |
| |
|
double | min_constraint () const |
| |
|
const data_size_t * | data_indices () const |
| | Get indices of data of current leaf.
|
| |
used to find split candidates for a leaf
◆ Init() [1/4]
| void LightGBM::LeafSplits::Init |
( |
const score_t * |
gradients, |
|
|
const score_t * |
hessians |
|
) |
| |
|
inline |
Init splits on current leaf, it will traverse all data to sum up the results.
- Parameters
-
◆ Init() [2/4]
| void LightGBM::LeafSplits::Init |
( |
double |
sum_gradients, |
|
|
double |
sum_hessians |
|
) |
| |
|
inline |
Init splits on current leaf, only update sum_gradients and sum_hessians.
- Parameters
-
| sum_gradients | |
| sum_hessians | |
◆ Init() [3/4]
| void LightGBM::LeafSplits::Init |
( |
int |
leaf, |
|
|
const DataPartition * |
data_partition, |
|
|
const score_t * |
gradients, |
|
|
const score_t * |
hessians |
|
) |
| |
|
inline |
Init splits on current leaf of partial data.
- Parameters
-
| leaf | Index of current leaf |
| data_partition | current data partition |
| gradients | |
| hessians | |
◆ Init() [4/4]
| void LightGBM::LeafSplits::Init |
( |
int |
leaf, |
|
|
const DataPartition * |
data_partition, |
|
|
double |
sum_gradients, |
|
|
double |
sum_hessians |
|
) |
| |
|
inline |
Init split on current leaf on partial data.
- Parameters
-
| leaf | Index of current leaf |
| data_partition | current data partition |
| sum_gradients | |
| sum_hessians | |
The documentation for this class was generated from the following file:
