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

Fix dataset error

Browse Source
This commit is contained in:
Aleksey Filippov 2025-03-11 23:35:23 +04:00
parent dc3d3620b7
commit 5817314562
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dt-cart/backend/service.py
View File

@ -15,8 +15,8 @@ def run_regression(
dataset_params: DatasetParams, dataset_params: DatasetParams,
tree_params: DecisionTreeParams, tree_params: DecisionTreeParams,
) -> RegressionResult: ) -> RegressionResult:
dataset: Dataset = Dataset(path=path, file=file)
try: try:
dataset: Dataset = Dataset(path=path, file=file)
data = dataset.read(dataset_params) data = dataset.read(dataset_params)
splitted_dataset: SplittedDataset = dataset.split_regression( splitted_dataset: SplittedDataset = dataset.split_regression(
data=data, data=data,
@ -38,8 +38,8 @@ def run_classification(
dataset_params: DatasetParams, dataset_params: DatasetParams,
tree_params: DecisionTreeParams, tree_params: DecisionTreeParams,
) -> ClassificationResult: ) -> ClassificationResult:
dataset: Dataset = Dataset(path=path, file=file)
try: try:
dataset: Dataset = Dataset(path=path, file=file)
data = dataset.read(dataset_params) data = dataset.read(dataset_params)
splitted_dataset: SplittedDataset = dataset.split_classification( splitted_dataset: SplittedDataset = dataset.split_classification(
data=data, data=data,