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.