Definition at line 108 of file RChunkLoader.hxx.
Public Member Functions | |
| RChunkLoader (ROOT::RDF::RNode &rdf, const std::size_t chunkSize, const std::size_t blockSize, const float validationSplit, const std::vector< std::string > &cols, const std::vector< std::size_t > &vecSizes={}, const float vecPadding=0.0, bool shuffle=true, const std::size_t setSeed=0) | |
| void | CheckIfOverlap (RFlat2DMatrix &Tensor1, RFlat2DMatrix &Tensor2) |
| void | CheckIfUnique (RFlat2DMatrix &Tensor) |
| void | CreateTrainingChunksIntervals () |
| Create training chunks consisiting of block intervals of different types. | |
| void | CreateValidationChunksIntervals () |
| Create training chunks consisiting of block intervals of different types. | |
| std::size_t | GetNumTrainingChunks () |
| std::size_t | GetNumTrainingEntries () |
| std::size_t | GetNumValidationChunks () |
| std::size_t | GetNumValidationEntries () |
| std::vector< std::size_t > | GetTrainingChunkSizes () |
| std::vector< std::size_t > | GetValidationChunkSizes () |
| void | LoadTrainingChunk (RFlat2DMatrix &TrainChunkTensor, std::size_t chunk) |
| Load the nth chunk from the training dataset into a tensor. | |
| void | LoadValidationChunk (RFlat2DMatrix &ValidationChunkTensor, std::size_t chunk) |
| Load the nth chunk from the validation dataset into a tensor. | |
| void | ResetDataframe () |
| void | SplitDataset () |
| Distribute the blocks into training and validation datasets. | |
Private Attributes | |
| ROOT::RDF::RNode & | f_rdf |
| std::size_t | fBlockSize |
| std::size_t | fChunkSize |
| std::vector< std::string > | fCols |
| ROOT::RDF::RResultPtr< std::vector< ULong64_t > > | fEntries |
| bool | fNotFiltered |
| std::size_t | fNumChunkCols |
| std::size_t | fNumCols |
| std::size_t | fNumEntries |
| std::size_t | fNumTrainEntries |
| std::size_t | fNumValidationEntries |
| std::size_t | fSetSeed |
| bool | fShuffle |
| std::size_t | fSumVecSizes |
| std::unique_ptr< RFlat2DMatrixOperators > | fTensorOperators |
| std::unique_ptr< RChunkConstructor > | fTraining |
| std::unique_ptr< RChunkConstructor > | fValidation |
| float | fValidationSplit |
| std::size_t | fVecPadding |
| std::vector< std::size_t > | fVecSizes |
#include <TMVA/BatchGenerator/RChunkLoader.hxx>
|
inline |
Definition at line 139 of file RChunkLoader.hxx.
|
inline |
Definition at line 450 of file RChunkLoader.hxx.
|
inline |
Definition at line 442 of file RChunkLoader.hxx.
|
inline |
Create training chunks consisiting of block intervals of different types.
Definition at line 252 of file RChunkLoader.hxx.
|
inline |
Create training chunks consisiting of block intervals of different types.
Definition at line 292 of file RChunkLoader.hxx.
|
inline |
Definition at line 473 of file RChunkLoader.hxx.
|
inline |
Definition at line 439 of file RChunkLoader.hxx.
|
inline |
Definition at line 475 of file RChunkLoader.hxx.
|
inline |
Definition at line 440 of file RChunkLoader.hxx.
|
inline |
Definition at line 436 of file RChunkLoader.hxx.
|
inline |
Definition at line 437 of file RChunkLoader.hxx.
|
inline |
Load the nth chunk from the training dataset into a tensor.
| [in] | TrainChunkTensor | RTensor for the training chunk |
| [in] | chunk | Index of the chunk in the dataset |
Definition at line 330 of file RChunkLoader.hxx.
|
inline |
Load the nth chunk from the validation dataset into a tensor.
| [in] | ValidationChunkTensor | RTensor for the validation chunk |
| [in] | chunk | Index of the chunk in the dataset |
Definition at line 383 of file RChunkLoader.hxx.
|
inline |
Definition at line 431 of file RChunkLoader.hxx.
|
inline |
Distribute the blocks into training and validation datasets.
Definition at line 177 of file RChunkLoader.hxx.
|
private |
Definition at line 125 of file RChunkLoader.hxx.
|
private |
Definition at line 113 of file RChunkLoader.hxx.
|
private |
Definition at line 112 of file RChunkLoader.hxx.
|
private |
Definition at line 126 of file RChunkLoader.hxx.
|
private |
Definition at line 133 of file RChunkLoader.hxx.
|
private |
Definition at line 130 of file RChunkLoader.hxx.
|
private |
Definition at line 119 of file RChunkLoader.hxx.
|
private |
Definition at line 127 of file RChunkLoader.hxx.
|
private |
Definition at line 111 of file RChunkLoader.hxx.
|
private |
Definition at line 121 of file RChunkLoader.hxx.
|
private |
Definition at line 122 of file RChunkLoader.hxx.
|
private |
Definition at line 128 of file RChunkLoader.hxx.
|
private |
Definition at line 131 of file RChunkLoader.hxx.
|
private |
Definition at line 117 of file RChunkLoader.hxx.
|
private |
Definition at line 123 of file RChunkLoader.hxx.
|
private |
Definition at line 135 of file RChunkLoader.hxx.
|
private |
Definition at line 136 of file RChunkLoader.hxx.
|
private |
Definition at line 114 of file RChunkLoader.hxx.
|
private |
Definition at line 118 of file RChunkLoader.hxx.
|
private |
Definition at line 116 of file RChunkLoader.hxx.