is.ulstu.ru/ds-services
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add default dataset target value

Browse Source
This commit is contained in:
Aleksey Filippov 2025-03-12 13:31:13 +04:00
parent f617f3e41c
commit bc0c132a35
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
dt-cart/backend/dataset/__init__.py
View File

@ -46,8 +46,9 @@ class Dataset:
random_state: int, random_state: int,
is_classification: bool = False, is_classification: bool = False,
) -> SplittedDataset: ) -> SplittedDataset:
target = params.target or data.columns[-1]
X = data.drop([params.target], axis=1) X = data.drop([params.target], axis=1)
y = data[[params.target]] y = data[[target]]
stratify = None if not is_classification else y stratify = None if not is_classification else y
X_train, X_test, y_train, y_test = train_test_split( X_train, X_test, y_train, y_test = train_test_split(
X, X,

2
dt-cart/backend/dataset/dto.py
View File

@ -8,7 +8,7 @@ class DatasetUploadDto(Schema):
class DatasetDto(Schema): class DatasetDto(Schema):
input = fields.List(fields.String(), load_default=None) input = fields.List(fields.String(), load_default=None)
target = fields.String(required=True) target = fields.String(load_default=None)
sep = fields.String(load_default=",") sep = fields.String(load_default=",")
decimal = 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)) train_volume = fields.Float(load_default=0.8, validate=Range(min=0.1, max=0.9))