StochTree 0.0.1
Loading...
Searching...
No Matches
Classes | Functions
Dataset API

Functions for loading, using, and modifying training / test data for forest samplers. More...

Classes

class  StochTree::ColumnMatrix
 Internal wrapper around Eigen::MatrixXd interface for multidimensional floating point data. More...
 
class  StochTree::ColumnVector
 Internal wrapper around Eigen::VectorXd interface for univariate floating point data. The (frequently updated) full / partial residual used in sampling forests is stored internally as a ColumnVector by the sampling functions (see Forest Sampler API). More...
 
class  StochTree::ForestDataset
 API for loading and accessing data used to sample tree ensembles The covariates / bases / weights used in sampling forests are stored internally as a ForestDataset by the sampling functions (see Forest Sampler API). More...
 
class  StochTree::RandomEffectsDataset
 API for loading and accessing data used to sample (additive) random effects. More...
 

Functions

static void StochTree::ExtractMultipleFeaturesFromMemory (std::vector< std::string > *text_data, const Parser *parser, std::vector< int32_t > &column_indices, Eigen::MatrixXd &data, data_size_t num_rows)
 Extract multiple features from the raw data loaded from a file into an Eigen::MatrixXd. Lightly modified from LightGBM's datasetloader interface to support stochtree's use cases.
 
static void StochTree::ExtractSingleFeatureFromMemory (std::vector< std::string > *text_data, const Parser *parser, int32_t column_index, Eigen::VectorXd &data, data_size_t num_rows)
 Extract a single feature from the raw data loaded from a file into an Eigen::VectorXd. Lightly modified from LightGBM's datasetloader interface to support stochtree's use cases.
 

Detailed Description

Functions for loading, using, and modifying training / test data for forest samplers.