Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
ROOT::Experimental::Internal::ML::RDatasetLoaderFunctor< ColTypes > Class Template Reference

template<typename... ColTypes>
class ROOT::Experimental::Internal::ML::RDatasetLoaderFunctor< ColTypes >

Loading chunks made in RDatasetLoader into tensors from data from RDataFrame.

Definition at line 34 of file RDatasetLoader.hxx.

Public Member Functions

 RDatasetLoaderFunctor (RFlat2DMatrix &datasetTensor, std::size_t numColumns, const std::vector< std::size_t > &maxVecSizes, float vecPadding, int i)
 
void operator() (const ColTypes &...cols)
 

Private Member Functions

template<typename T , std::enable_if_t<!ROOT::Internal::RDF::IsDataContainer< T >::value, int > = 0>
void AssignToTensor (const T &val, int i, int numColumns)
 Copy the content of a column into RTensor when the column consits of single values.
 
template<typename T , std::enable_if_t< ROOT::Internal::RDF::IsDataContainer< T >::value, int > = 0>
void AssignToTensor (const T &vec, int i, int numColumns)
 Copy the content of a column into RTensor when the column consits of vectors.
 

Private Attributes

RFlat2DMatrixfDatasetTensor
 
int fI
 
std::vector< std::size_t > fMaxVecSizes {}
 
int fNumColumns
 
std::size_t fNumDatasetCols
 
std::size_t fOffset {}
 
float fVecPadding {}
 
std::size_t fVecSizeIdx {}
 

#include <ROOT/ML/RDatasetLoader.hxx>

Constructor & Destructor Documentation

◆ RDatasetLoaderFunctor()

template<typename... ColTypes>
ROOT::Experimental::Internal::ML::RDatasetLoaderFunctor< ColTypes >::RDatasetLoaderFunctor ( RFlat2DMatrix & datasetTensor,
std::size_t numColumns,
const std::vector< std::size_t > & maxVecSizes,
float vecPadding,
int i )
inline

Definition at line 75 of file RDatasetLoader.hxx.

Member Function Documentation

◆ AssignToTensor() [1/2]

template<typename... ColTypes>
template<typename T , std::enable_if_t<!ROOT::Internal::RDF::IsDataContainer< T >::value, int > = 0>
void ROOT::Experimental::Internal::ML::RDatasetLoaderFunctor< ColTypes >::AssignToTensor ( const T & val,
int i,
int numColumns )
inlineprivate

Copy the content of a column into RTensor when the column consits of single values.

Definition at line 68 of file RDatasetLoader.hxx.

◆ AssignToTensor() [2/2]

template<typename... ColTypes>
template<typename T , std::enable_if_t< ROOT::Internal::RDF::IsDataContainer< T >::value, int > = 0>
void ROOT::Experimental::Internal::ML::RDatasetLoaderFunctor< ColTypes >::AssignToTensor ( const T & vec,
int i,
int numColumns )
inlineprivate

Copy the content of a column into RTensor when the column consits of vectors.

Definition at line 49 of file RDatasetLoader.hxx.

◆ operator()()

template<typename... ColTypes>
void ROOT::Experimental::Internal::ML::RDatasetLoaderFunctor< ColTypes >::operator() ( const ColTypes &... cols)
inline

Definition at line 85 of file RDatasetLoader.hxx.

Member Data Documentation

◆ fDatasetTensor

template<typename... ColTypes>
RFlat2DMatrix& ROOT::Experimental::Internal::ML::RDatasetLoaderFunctor< ColTypes >::fDatasetTensor
private

Definition at line 39 of file RDatasetLoader.hxx.

◆ fI

template<typename... ColTypes>
int ROOT::Experimental::Internal::ML::RDatasetLoaderFunctor< ColTypes >::fI
private

Definition at line 43 of file RDatasetLoader.hxx.

◆ fMaxVecSizes

template<typename... ColTypes>
std::vector<std::size_t> ROOT::Experimental::Internal::ML::RDatasetLoaderFunctor< ColTypes >::fMaxVecSizes {}
private

Definition at line 38 of file RDatasetLoader.hxx.

◆ fNumColumns

template<typename... ColTypes>
int ROOT::Experimental::Internal::ML::RDatasetLoaderFunctor< ColTypes >::fNumColumns
private

Definition at line 44 of file RDatasetLoader.hxx.

◆ fNumDatasetCols

template<typename... ColTypes>
std::size_t ROOT::Experimental::Internal::ML::RDatasetLoaderFunctor< ColTypes >::fNumDatasetCols
private

Definition at line 41 of file RDatasetLoader.hxx.

◆ fOffset

template<typename... ColTypes>
std::size_t ROOT::Experimental::Internal::ML::RDatasetLoaderFunctor< ColTypes >::fOffset {}
private

Definition at line 35 of file RDatasetLoader.hxx.

◆ fVecPadding

template<typename... ColTypes>
float ROOT::Experimental::Internal::ML::RDatasetLoaderFunctor< ColTypes >::fVecPadding {}
private

Definition at line 37 of file RDatasetLoader.hxx.

◆ fVecSizeIdx

template<typename... ColTypes>
std::size_t ROOT::Experimental::Internal::ML::RDatasetLoaderFunctor< ColTypes >::fVecSizeIdx {}
private

Definition at line 36 of file RDatasetLoader.hxx.

  • tree/ml/inc/ROOT/ML/RDatasetLoader.hxx