Regression Configs
Dummy Regression
- class InnerEye.ML.configs.regression.DummyRegression.DummyRegression[source]
Config file for a dummy regression model
- create_model() Any[source]
Creates a PyTorch model from the settings stored in the present object. This is an abstract method that each model class (segmentation, regression) should override. Return type is LightningModule, not Any - but we want to avoid importing torch at this point.
- get_label_transform() Union[Callable, List[Callable]][source]
Return a transformation or list of transformation to apply to the labels.
- get_model_train_test_dataset_splits(dataset_df: DataFrame) DatasetSplits[source]
Computes the training, validation and test splits for the model, from a dataframe that contains the full dataset.
- Parameters:
dataset_df – A dataframe that contains the full dataset that the model is using.
- Returns:
An instance of DatasetSplits with dataframes for training, validation and testing.
- name = 'DummyRegression'
- param = <param.parameterized.Parameters object>