DataModule.summary_input_output_dimensions

DataModule.summary_input_output_dimensions(print_summary: bool = True) Tuple[int, int, str][source]

Calculates the dimensions of the input and output of the ML model.

Parameters:

print_summarybool

Whether to print the summary text.

Returns:

input_ml_totalint

Total number of dimensions of the input of the ML model.

output_ml_totalint

Total number of dimensions of the output of the ML model.

summarystr

Summary text.