Logo ROOT  

Reference Guide
 
Loading...
Searching...
No Matches
List of all members | Public Member Functions | Private Attributes | List of all members
TMVA::Experimental::Internal::RDatasetLoader< Args > Class Template Reference

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

Definition at line 103 of file RDatasetLoader.hxx.

Public Member Functions

 RDatasetLoader (const std::vector< ROOT::RDF::RNode > &rdfs, 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)
 
std::vector< RFlat2DMatrix > GetTrainingDatasets ()
 
std::vector< RFlat2DMatrix > GetValidationDatasets ()
 
void SplitDataframe (ROOT::RDF::RNode &rdf, RFlat2DMatrix &TrainingDataset, RFlat2DMatrix &ValidationDataset)
 Split an individual dataframe into a training and validation dataset.
 
void SplitDatasets ()
 Split the dataframes in a training and validation dataset.
 

Private Attributes

std::vector< ROOT::RDF::RNode > f_rdfs
 
std::vector< std::string > fCols
 
ROOT::RDF::RResultPtr< std::vector< ULong64_t > > fEntries
 
bool fNotFiltered
 
std::size_t fNumCols
 
std::size_t fNumDatasetCols
 
std::size_t fNumEntries
 
std::size_t fNumTrainingEntries
 
std::size_t fNumValidationEntries
 
std::size_t fSetSeed
 
bool fShuffle
 
std::size_t fSumVecSizes
 
std::unique_ptr< RFlat2DMatrixOperators > fTensorOperators
 
std::vector< RFlat2DMatrix > fTrainingDatasets
 
std::vector< RFlat2DMatrix > fValidationDatasets
 
float fValidationSplit
 
std::size_t fVecPadding
 
std::vector< std::size_t > fVecSizes
 

#include <TMVA/BatchGenerator/RDatasetLoader.hxx>

Constructor & Destructor Documentation

◆ RDatasetLoader()

template<typename... Args>
TMVA::Experimental::Internal::RDatasetLoader< Args >::RDatasetLoader ( const std::vector< ROOT::RDF::RNode > & rdfs,
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 132 of file RDatasetLoader.hxx.

Member Function Documentation

◆ GetTrainingDatasets()

template<typename... Args>
std::vector< RFlat2DMatrix > TMVA::Experimental::Internal::RDatasetLoader< Args >::GetTrainingDatasets ( )
inline

Definition at line 212 of file RDatasetLoader.hxx.

◆ GetValidationDatasets()

template<typename... Args>
std::vector< RFlat2DMatrix > TMVA::Experimental::Internal::RDatasetLoader< Args >::GetValidationDatasets ( )
inline

Definition at line 213 of file RDatasetLoader.hxx.

◆ SplitDataframe()

template<typename... Args>
void TMVA::Experimental::Internal::RDatasetLoader< Args >::SplitDataframe ( ROOT::RDF::RNode & rdf,
RFlat2DMatrix & TrainingDataset,
RFlat2DMatrix & ValidationDataset )
inline

Split an individual dataframe into a training and validation dataset.

Parameters
[in]rdfDataframe that will be split into training and validation
[in]TrainingDatasetTensor for the training dataset
[in]ValidationDatasetTensor for the validation dataset

Definition at line 155 of file RDatasetLoader.hxx.

◆ SplitDatasets()

template<typename... Args>
void TMVA::Experimental::Internal::RDatasetLoader< Args >::SplitDatasets ( )
inline

Split the dataframes in a training and validation dataset.

Definition at line 192 of file RDatasetLoader.hxx.

Member Data Documentation

◆ f_rdfs

template<typename... Args>
std::vector<ROOT::RDF::RNode> TMVA::Experimental::Internal::RDatasetLoader< Args >::f_rdfs
private

Definition at line 121 of file RDatasetLoader.hxx.

◆ fCols

template<typename... Args>
std::vector<std::string> TMVA::Experimental::Internal::RDatasetLoader< Args >::fCols
private

Definition at line 122 of file RDatasetLoader.hxx.

◆ fEntries

template<typename... Args>
ROOT::RDF::RResultPtr<std::vector<ULong64_t> > TMVA::Experimental::Internal::RDatasetLoader< Args >::fEntries
private

Definition at line 129 of file RDatasetLoader.hxx.

◆ fNotFiltered

template<typename... Args>
bool TMVA::Experimental::Internal::RDatasetLoader< Args >::fNotFiltered
private

Definition at line 126 of file RDatasetLoader.hxx.

◆ fNumCols

template<typename... Args>
std::size_t TMVA::Experimental::Internal::RDatasetLoader< Args >::fNumCols
private

Definition at line 123 of file RDatasetLoader.hxx.

◆ fNumDatasetCols

template<typename... Args>
std::size_t TMVA::Experimental::Internal::RDatasetLoader< Args >::fNumDatasetCols
private

Definition at line 112 of file RDatasetLoader.hxx.

◆ fNumEntries

template<typename... Args>
std::size_t TMVA::Experimental::Internal::RDatasetLoader< Args >::fNumEntries
private

Definition at line 106 of file RDatasetLoader.hxx.

◆ fNumTrainingEntries

template<typename... Args>
std::size_t TMVA::Experimental::Internal::RDatasetLoader< Args >::fNumTrainingEntries
private

Definition at line 117 of file RDatasetLoader.hxx.

◆ fNumValidationEntries

template<typename... Args>
std::size_t TMVA::Experimental::Internal::RDatasetLoader< Args >::fNumValidationEntries
private

Definition at line 118 of file RDatasetLoader.hxx.

◆ fSetSeed

template<typename... Args>
std::size_t TMVA::Experimental::Internal::RDatasetLoader< Args >::fSetSeed
private

Definition at line 124 of file RDatasetLoader.hxx.

◆ fShuffle

template<typename... Args>
bool TMVA::Experimental::Internal::RDatasetLoader< Args >::fShuffle
private

Definition at line 127 of file RDatasetLoader.hxx.

◆ fSumVecSizes

template<typename... Args>
std::size_t TMVA::Experimental::Internal::RDatasetLoader< Args >::fSumVecSizes
private

Definition at line 110 of file RDatasetLoader.hxx.

◆ fTensorOperators

template<typename... Args>
std::unique_ptr<RFlat2DMatrixOperators> TMVA::Experimental::Internal::RDatasetLoader< Args >::fTensorOperators
private

Definition at line 119 of file RDatasetLoader.hxx.

◆ fTrainingDatasets

template<typename... Args>
std::vector<RFlat2DMatrix> TMVA::Experimental::Internal::RDatasetLoader< Args >::fTrainingDatasets
private

Definition at line 114 of file RDatasetLoader.hxx.

◆ fValidationDatasets

template<typename... Args>
std::vector<RFlat2DMatrix> TMVA::Experimental::Internal::RDatasetLoader< Args >::fValidationDatasets
private

Definition at line 115 of file RDatasetLoader.hxx.

◆ fValidationSplit

template<typename... Args>
float TMVA::Experimental::Internal::RDatasetLoader< Args >::fValidationSplit
private

Definition at line 107 of file RDatasetLoader.hxx.

◆ fVecPadding

template<typename... Args>
std::size_t TMVA::Experimental::Internal::RDatasetLoader< Args >::fVecPadding
private

Definition at line 111 of file RDatasetLoader.hxx.

◆ fVecSizes

template<typename... Args>
std::vector<std::size_t> TMVA::Experimental::Internal::RDatasetLoader< Args >::fVecSizes
private

Definition at line 109 of file RDatasetLoader.hxx.

  • tmva/tmva/inc/TMVA/BatchGenerator/RDatasetLoader.hxx