Definition at line 33 of file DataLoader.h.
Public Member Functions | |
TDataLoader (const AData &data, size_t nSamples, size_t batchSize, size_t nInputFeatures, size_t nOutputFeatures, size_t nthreads=1) | |
TDataLoader (const TDataLoader &)=default | |
TDataLoader (TDataLoader &&)=default | |
BatchIterator_t | begin () |
void | CopyInput (TMatrixT< AReal > &matrix, IndexIterator_t begin) |
Copy input matrix into the given host buffer. | |
void | CopyOutput (TMatrixT< AReal > &matrix, IndexIterator_t begin) |
Copy output matrix into the given host buffer. | |
void | CopyWeights (TMatrixT< AReal > &matrix, IndexIterator_t begin) |
Copy weight matrix into the given host buffer. | |
BatchIterator_t | end () |
TBatch< TReference< AReal > > | GetBatch () |
Return the next batch from the training set. | |
TDataLoader & | operator= (const TDataLoader &)=default |
TDataLoader & | operator= (TDataLoader &&)=default |
void | Shuffle () |
Shuffle the order of the samples in the batch. | |
Private Types | |
using | BatchIterator_t = TBatchIterator< AData, TReference< AReal > > |
Private Attributes | |
size_t | fBatchIndex |
size_t | fBatchSize |
const AData & | fData |
size_t | fNInputFeatures |
size_t | fNOutputFeatures |
size_t | fNSamples |
std::vector< size_t > | fSampleIndices |
Ordering of the samples in the epoch. | |
TMatrixT< AReal > | inputMatrix |
TMatrixT< AReal > | outputMatrix |
TMatrixT< AReal > | weightMatrix |
#include <TMVA/DNN/Architectures/Reference/DataLoader.h>
|
private |
Definition at line 35 of file DataLoader.h.
TMVA::DNN::TDataLoader< AData, TReference< AReal > >::TDataLoader | ( | const AData & | data, |
size_t | nSamples, | ||
size_t | batchSize, | ||
size_t | nInputFeatures, | ||
size_t | nOutputFeatures, | ||
size_t | nthreads = 1 |
||
) |
Definition at line 84 of file DataLoader.h.
|
default |
|
default |
|
inline |
Definition at line 69 of file DataLoader.h.
void TMVA::DNN::TDataLoader< AData, TReference< AReal > >::CopyInput | ( | TMatrixT< AReal > & | matrix, |
IndexIterator_t | begin | ||
) |
Copy input matrix into the given host buffer.
Function to be specialized by the architecture-specific backend.
void TMVA::DNN::TDataLoader< AData, TReference< AReal > >::CopyOutput | ( | TMatrixT< AReal > & | matrix, |
IndexIterator_t | begin | ||
) |
Copy output matrix into the given host buffer.
Function to be specialized by the architecture-specific backend.
void TMVA::DNN::TDataLoader< AData, TReference< AReal > >::CopyWeights | ( | TMatrixT< AReal > & | matrix, |
IndexIterator_t | begin | ||
) |
Copy weight matrix into the given host buffer.
Function to be specialized by the architecture-specific backend.
|
inline |
Definition at line 70 of file DataLoader.h.
TBatch< TReference< AReal > > TMVA::DNN::TDataLoader< AData, TReference< AReal > >::GetBatch |
Return the next batch from the training set.
The TDataLoader object keeps an internal counter that cycles over the batches in the training set.
Definition at line 97 of file DataLoader.h.
|
default |
|
default |
void TMVA::DNN::TDataLoader< AData, TReference< AReal > >::Shuffle |
Shuffle the order of the samples in the batch.
The shuffling is indirect, i.e. only the indices are shuffled. No input data is moved by this routine.
Definition at line 115 of file DataLoader.h.
|
private |
Definition at line 43 of file DataLoader.h.
|
private |
Definition at line 40 of file DataLoader.h.
|
private |
Definition at line 37 of file DataLoader.h.
|
private |
Definition at line 41 of file DataLoader.h.
|
private |
Definition at line 42 of file DataLoader.h.
|
private |
Definition at line 39 of file DataLoader.h.
|
private |
Ordering of the samples in the epoch.
Definition at line 49 of file DataLoader.h.
|
private |
Definition at line 45 of file DataLoader.h.
|
private |
Definition at line 46 of file DataLoader.h.
|
private |
Definition at line 47 of file DataLoader.h.