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

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

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

Definition at line 37 of file RChunkLoader.hxx.

Public Member Functions

 RChunkLoaderFunctor (RFlat2DMatrix &chunkTensor, 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

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

#include <ROOT/ML/RChunkLoader.hxx>

Constructor & Destructor Documentation

◆ RChunkLoaderFunctor()

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

Definition at line 80 of file RChunkLoader.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::RChunkLoaderFunctor< 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 72 of file RChunkLoader.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::RChunkLoaderFunctor< 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 52 of file RChunkLoader.hxx.

◆ operator()()

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

Definition at line 86 of file RChunkLoader.hxx.

Member Data Documentation

◆ fChunkTensor

template<typename... ColTypes>
RFlat2DMatrix& ROOT::Experimental::Internal::ML::RChunkLoaderFunctor< ColTypes >::fChunkTensor
private

Definition at line 42 of file RChunkLoader.hxx.

◆ fI

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

Definition at line 46 of file RChunkLoader.hxx.

◆ fMaxVecSizes

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

Definition at line 41 of file RChunkLoader.hxx.

◆ fNumChunkCols

template<typename... ColTypes>
std::size_t ROOT::Experimental::Internal::ML::RChunkLoaderFunctor< ColTypes >::fNumChunkCols
private

Definition at line 44 of file RChunkLoader.hxx.

◆ fNumColumns

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

Definition at line 47 of file RChunkLoader.hxx.

◆ fOffset

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

Definition at line 38 of file RChunkLoader.hxx.

◆ fVecPadding

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

Definition at line 40 of file RChunkLoader.hxx.

◆ fVecSizeIdx

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

Definition at line 39 of file RChunkLoader.hxx.

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