Writes batches into an RNTuple, as one float field per scalar column and one std::vector<float> field per vector column.
Definition at line 100 of file RBatchSink.hxx.
Public Member Functions | |
| RNTupleBatchSink (const RNTupleBatchSink &)=delete | |
| RNTupleBatchSink (RNTupleBatchSink &&)=delete | |
| RNTupleBatchSink (std::string_view dataset_name, std::string_view filename, std::vector< RColumnLayout > layout) | |
| ~RNTupleBatchSink () override | |
| void | Commit () override |
| Flush everything to disk once after the last batch. | |
| void | FillBatch (const RFlat2DMatrix &batch) |
Write every row of batch. | |
| void | FillRow (const float *row) override |
Copy one row of fRowWidth floats into the writer's buffers and write the entry. | |
| RNTupleBatchSink & | operator= (const RNTupleBatchSink &)=delete |
| RNTupleBatchSink & | operator= (RNTupleBatchSink &&)=delete |
Protected Attributes | |
| std::vector< RColumnLayout > | fLayout |
| std::size_t | fRowWidth {} |
Private Attributes | |
| std::vector< float * > | fDestinations |
| std::vector< std::shared_ptr< float > > | fScalarFields |
| std::vector< std::shared_ptr< std::vector< float > > > | fVectorFields |
| std::unique_ptr< ROOT::RNTupleWriter > | fWriter |
#include <ROOT/ML/RBatchSink.hxx>
| ROOT::Experimental::Internal::ML::RNTupleBatchSink::RNTupleBatchSink | ( | std::string_view | dataset_name, |
| std::string_view | filename, | ||
| std::vector< RColumnLayout > | layout ) |
Definition at line 79 of file RBatchSink.cxx.
|
delete |
|
delete |
|
overridedefault |
|
overridevirtual |
Flush everything to disk once after the last batch.
Implements ROOT::Experimental::Internal::ML::RBatchSink.
Definition at line 111 of file RBatchSink.cxx.
|
inherited |
Write every row of batch.
Definition at line 25 of file RBatchSink.cxx.
|
overridevirtual |
Copy one row of fRowWidth floats into the writer's buffers and write the entry.
Implements ROOT::Experimental::Internal::ML::RBatchSink.
Definition at line 104 of file RBatchSink.cxx.
|
delete |
|
delete |
|
private |
Definition at line 103 of file RBatchSink.hxx.
|
protectedinherited |
Definition at line 52 of file RBatchSink.hxx.
|
protectedinherited |
Definition at line 53 of file RBatchSink.hxx.
|
private |
Definition at line 101 of file RBatchSink.hxx.
|
private |
Definition at line 102 of file RBatchSink.hxx.
|
private |
Definition at line 104 of file RBatchSink.hxx.