template<typename... Args>
class TMVA::Experimental::Internal::RChunkLoader< Args >

Definition at line 107 of file RChunkLoader.hxx.

Public Member Functions
	RChunkLoader (ROOT::RDF::RNode &rdf, std::size_t numEntries, ROOT::RDF::RResultPtr< std::vector< ULong64_t > > rdf_entries, 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 (TMVA::Experimental::RTensor< float > &Tensor1, TMVA::Experimental::RTensor< float > &Tensor2)

void	CheckIfUnique (TMVA::Experimental::RTensor< float > &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 (TMVA::Experimental::RTensor< float > &TrainChunkTensor, std::size_t chunk)
	Load the nth chunk from the training dataset into a tensor.

void	LoadValidationChunk (TMVA::Experimental::RTensor< float > &ValidationChunkTensor, std::size_t chunk)
	Load the nth chunk from the validation dataset into a tensor.

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< RChunkConstructor >	fTraining

std::unique_ptr< RChunkConstructor >	fValidation

float	fValidationSplit

std::size_t	fVecPadding

std::vector< std::size_t >	fVecSizes

#include <TMVA/BatchGenerator/RChunkLoader.hxx>

Constructor & Destructor Documentation

◆ RChunkLoader()

template<typename... Args>

TMVA::Experimental::Internal::RChunkLoader< Args >::RChunkLoader	(	ROOT::RDF::RNode &	rdf,
		std::size_t	numEntries,
		ROOT::RDF::RResultPtr< std::vector< ULong64_t > >	rdf_entries,
		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 )

inline

Definition at line 137 of file RChunkLoader.hxx.

Member Function Documentation

◆ CheckIfOverlap()

template<typename... Args>

void TMVA::Experimental::Internal::RChunkLoader< Args >::CheckIfOverlap	(	TMVA::Experimental::RTensor< float > &	Tensor1,
		TMVA::Experimental::RTensor< float > &	Tensor2 )

inline

Definition at line 486 of file RChunkLoader.hxx.

◆ CheckIfUnique()

template<typename... Args>

void TMVA::Experimental::Internal::RChunkLoader< Args >::CheckIfUnique ( TMVA::Experimental::RTensor< float > & Tensor )

inline

Definition at line 470 of file RChunkLoader.hxx.

◆ CreateTrainingChunksIntervals()

template<typename... Args>

void TMVA::Experimental::Internal::RChunkLoader< Args >::CreateTrainingChunksIntervals ( )

inline

Create training chunks consisiting of block intervals of different types.

Definition at line 245 of file RChunkLoader.hxx.

◆ CreateValidationChunksIntervals()

template<typename... Args>

void TMVA::Experimental::Internal::RChunkLoader< Args >::CreateValidationChunksIntervals ( )

inline

Create training chunks consisiting of block intervals of different types.

Definition at line 285 of file RChunkLoader.hxx.

◆ GetNumTrainingChunks()

template<typename... Args>

std::size_t TMVA::Experimental::Internal::RChunkLoader< Args >::GetNumTrainingChunks ( )

inline

Definition at line 525 of file RChunkLoader.hxx.

◆ GetNumTrainingEntries()

template<typename... Args>

std::size_t TMVA::Experimental::Internal::RChunkLoader< Args >::GetNumTrainingEntries ( )

inline

Definition at line 467 of file RChunkLoader.hxx.

◆ GetNumValidationChunks()

template<typename... Args>

std::size_t TMVA::Experimental::Internal::RChunkLoader< Args >::GetNumValidationChunks ( )

inline

Definition at line 527 of file RChunkLoader.hxx.

◆ GetNumValidationEntries()

template<typename... Args>

std::size_t TMVA::Experimental::Internal::RChunkLoader< Args >::GetNumValidationEntries ( )

inline

Definition at line 468 of file RChunkLoader.hxx.

◆ GetTrainingChunkSizes()

template<typename... Args>

std::vector< std::size_t > TMVA::Experimental::Internal::RChunkLoader< Args >::GetTrainingChunkSizes ( )

inline

Definition at line 464 of file RChunkLoader.hxx.

◆ GetValidationChunkSizes()

template<typename... Args>

std::vector< std::size_t > TMVA::Experimental::Internal::RChunkLoader< Args >::GetValidationChunkSizes ( )

inline

Definition at line 465 of file RChunkLoader.hxx.

◆ LoadTrainingChunk()

template<typename... Args>

void TMVA::Experimental::Internal::RChunkLoader< Args >::LoadTrainingChunk	(	TMVA::Experimental::RTensor< float > &	TrainChunkTensor,
		std::size_t	chunk )

inline

Load the nth chunk from the training dataset into a tensor.

Parameters

[in]	TrainChunkTensor	RTensor for the training chunk
[in]	chunk	Index of the chunk in the dataset

Definition at line 323 of file RChunkLoader.hxx.

◆ LoadValidationChunk()

template<typename... Args>

void TMVA::Experimental::Internal::RChunkLoader< Args >::LoadValidationChunk	(	TMVA::Experimental::RTensor< float > &	ValidationChunkTensor,
		std::size_t	chunk )

inline

Load the nth chunk from the validation dataset into a tensor.

Parameters

[in]	ValidationChunkTensor	RTensor for the validation chunk
[in]	chunk	Index of the chunk in the dataset

Definition at line 396 of file RChunkLoader.hxx.

◆ SplitDataset()

template<typename... Args>

void TMVA::Experimental::Internal::RChunkLoader< Args >::SplitDataset ( )

inline

Distribute the blocks into training and validation datasets.

Definition at line 170 of file RChunkLoader.hxx.

Member Data Documentation

◆ f_rdf

template<typename... Args>

ROOT::RDF::RNode& TMVA::Experimental::Internal::RChunkLoader< Args >::f_rdf

private

Definition at line 123 of file RChunkLoader.hxx.

◆ fBlockSize

template<typename... Args>

std::size_t TMVA::Experimental::Internal::RChunkLoader< Args >::fBlockSize

private

Definition at line 112 of file RChunkLoader.hxx.

◆ fChunkSize

template<typename... Args>

std::size_t TMVA::Experimental::Internal::RChunkLoader< Args >::fChunkSize

private

Definition at line 111 of file RChunkLoader.hxx.

◆ fCols

template<typename... Args>

std::vector<std::string> TMVA::Experimental::Internal::RChunkLoader< Args >::fCols

private

Definition at line 124 of file RChunkLoader.hxx.

◆ fEntries

template<typename... Args>

ROOT::RDF::RResultPtr<std::vector<ULong64_t> > TMVA::Experimental::Internal::RChunkLoader< Args >::fEntries

private

Definition at line 131 of file RChunkLoader.hxx.

◆ fNotFiltered

template<typename... Args>

bool TMVA::Experimental::Internal::RChunkLoader< Args >::fNotFiltered

private

Definition at line 128 of file RChunkLoader.hxx.

◆ fNumChunkCols

template<typename... Args>

std::size_t TMVA::Experimental::Internal::RChunkLoader< Args >::fNumChunkCols

private

Definition at line 118 of file RChunkLoader.hxx.

◆ fNumCols

template<typename... Args>

std::size_t TMVA::Experimental::Internal::RChunkLoader< Args >::fNumCols

private

Definition at line 125 of file RChunkLoader.hxx.

◆ fNumEntries

template<typename... Args>

std::size_t TMVA::Experimental::Internal::RChunkLoader< Args >::fNumEntries

private

Definition at line 110 of file RChunkLoader.hxx.

◆ fNumTrainEntries

template<typename... Args>

std::size_t TMVA::Experimental::Internal::RChunkLoader< Args >::fNumTrainEntries

private

Definition at line 120 of file RChunkLoader.hxx.

◆ fNumValidationEntries

template<typename... Args>

std::size_t TMVA::Experimental::Internal::RChunkLoader< Args >::fNumValidationEntries

private

Definition at line 121 of file RChunkLoader.hxx.

◆ fSetSeed

template<typename... Args>

std::size_t TMVA::Experimental::Internal::RChunkLoader< Args >::fSetSeed

private

Definition at line 126 of file RChunkLoader.hxx.

◆ fShuffle

template<typename... Args>

bool TMVA::Experimental::Internal::RChunkLoader< Args >::fShuffle

private

Definition at line 129 of file RChunkLoader.hxx.

◆ fSumVecSizes

template<typename... Args>

std::size_t TMVA::Experimental::Internal::RChunkLoader< Args >::fSumVecSizes

private

Definition at line 116 of file RChunkLoader.hxx.

◆ fTraining

template<typename... Args>

std::unique_ptr<RChunkConstructor> TMVA::Experimental::Internal::RChunkLoader< Args >::fTraining

private

Definition at line 133 of file RChunkLoader.hxx.

◆ fValidation

template<typename... Args>

std::unique_ptr<RChunkConstructor> TMVA::Experimental::Internal::RChunkLoader< Args >::fValidation

private

Definition at line 134 of file RChunkLoader.hxx.

◆ fValidationSplit

template<typename... Args>

float TMVA::Experimental::Internal::RChunkLoader< Args >::fValidationSplit

private

Definition at line 113 of file RChunkLoader.hxx.

◆ fVecPadding

template<typename... Args>

std::size_t TMVA::Experimental::Internal::RChunkLoader< Args >::fVecPadding

private

Definition at line 117 of file RChunkLoader.hxx.

◆ fVecSizes

template<typename... Args>

std::vector<std::size_t> TMVA::Experimental::Internal::RChunkLoader< Args >::fVecSizes

private

Definition at line 115 of file RChunkLoader.hxx.

tmva/tmva/inc/TMVA/BatchGenerator/RChunkLoader.hxx

Public Member Functions

Private Attributes

Constructor & Destructor Documentation

◆ RChunkLoader()

Member Function Documentation

◆ CheckIfOverlap()

◆ CheckIfUnique()

◆ CreateTrainingChunksIntervals()

◆ CreateValidationChunksIntervals()

◆ GetNumTrainingChunks()

◆ GetNumTrainingEntries()

◆ GetNumValidationChunks()

◆ GetNumValidationEntries()

◆ GetTrainingChunkSizes()

◆ GetValidationChunkSizes()

◆ LoadTrainingChunk()

◆ LoadValidationChunk()

◆ SplitDataset()

Member Data Documentation

◆ f_rdf

◆ fBlockSize

◆ fChunkSize

◆ fCols

◆ fEntries

◆ fNotFiltered

◆ fNumChunkCols

◆ fNumCols

◆ fNumEntries

◆ fNumTrainEntries

◆ fNumValidationEntries

◆ fSetSeed

◆ fShuffle

◆ fSumVecSizes

◆ fTraining

◆ fValidation

◆ fValidationSplit

◆ fVecPadding

◆ fVecSizes