mlrl.testbed.experiments.output.sinks.sink_csv module

Author Michael Rapp (michael.rapp.ml@gmail.com)

Provides classes that allow writing output data to CSV files.

class mlrl.testbed.experiments.output.sinks.sink_csv.CsvFileSink(directory: Path, options: Options = <mlrl.util.options.Options object>, create_directory: bool = False)

Bases: TabularFileSink

Allows to write tabular output data to a CSV file.

COLUMN_MODEL_SIZE = 'Model size'
create_source(input_directory: Path) Source | None

Must be implemented by subclasses in order to create a Source that can read the data written to this sink.

Returns:

The Source that has been created or None, if no source is available