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>