diff --git a/dt-cart/backend/dataset/__init__.py b/dt-cart/backend/dataset/__init__.py index 1d3330b..4d40140 100644 --- a/dt-cart/backend/dataset/__init__.py +++ b/dt-cart/backend/dataset/__init__.py @@ -46,8 +46,9 @@ class Dataset: random_state: int, is_classification: bool = False, ) -> SplittedDataset: + target = params.target or data.columns[-1] X = data.drop([params.target], axis=1) - y = data[[params.target]] + y = data[[target]] stratify = None if not is_classification else y X_train, X_test, y_train, y_test = train_test_split( X, diff --git a/dt-cart/backend/dataset/dto.py b/dt-cart/backend/dataset/dto.py index 3927bf2..8fe04e2 100644 --- a/dt-cart/backend/dataset/dto.py +++ b/dt-cart/backend/dataset/dto.py @@ -8,7 +8,7 @@ class DatasetUploadDto(Schema): class DatasetDto(Schema): input = fields.List(fields.String(), load_default=None) - target = fields.String(required=True) + target = fields.String(load_default=None) sep = fields.String(load_default=",") decimal = fields.String(load_default=".") train_volume = fields.Float(load_default=0.8, validate=Range(min=0.1, max=0.9))