Writes batches into a TTree, as one float leaf per scalar column and one std::vector<float> branch per vector column, matching the RNTuple sink's schema.
Definition at line 80 of file RBatchSink.hxx.
Public Member Functions | |
| RTTreeBatchSink (const RTTreeBatchSink &)=delete | |
| RTTreeBatchSink (RTTreeBatchSink &&)=delete | |
| RTTreeBatchSink (std::string_view dataset_name, std::string_view filename, std::vector< RColumnLayout > layout) | |
| ~RTTreeBatchSink () 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. | |
| RTTreeBatchSink & | operator= (const RTTreeBatchSink &)=delete |
| RTTreeBatchSink & | operator= (RTTreeBatchSink &&)=delete |
Protected Attributes | |
| std::vector< RColumnLayout > | fLayout |
| std::size_t | fRowWidth {} |
Private Attributes | |
| std::unique_ptr< TFile > | fFile |
| std::vector< float > | fRow |
| TTree * | fTree |
| std::vector< std::vector< float > > | fVectorBuffers |
#include <ROOT/ML/RBatchSink.hxx>
| ROOT::Experimental::Internal::ML::RTTreeBatchSink::RTTreeBatchSink | ( | std::string_view | dataset_name, |
| std::string_view | filename, | ||
| std::vector< RColumnLayout > | layout ) |
Definition at line 36 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 74 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 61 of file RBatchSink.cxx.
|
delete |
|
delete |
|
private |
Definition at line 81 of file RBatchSink.hxx.
|
protectedinherited |
Definition at line 52 of file RBatchSink.hxx.
|
private |
Definition at line 83 of file RBatchSink.hxx.
|
protectedinherited |
Definition at line 53 of file RBatchSink.hxx.
|
private |
Definition at line 82 of file RBatchSink.hxx.
|
private |
Definition at line 84 of file RBatchSink.hxx.