diff --git a/dt-cart/backend/service.py b/dt-cart/backend/service.py index 2c824c0..6aec393 100644 --- a/dt-cart/backend/service.py +++ b/dt-cart/backend/service.py @@ -15,8 +15,8 @@ def run_regression( dataset_params: DatasetParams, tree_params: DecisionTreeParams, ) -> RegressionResult: + dataset: Dataset = Dataset(path=path, file=file) try: - dataset: Dataset = Dataset(path=path, file=file) data = dataset.read(dataset_params) splitted_dataset: SplittedDataset = dataset.split_regression( data=data, @@ -38,8 +38,8 @@ def run_classification( dataset_params: DatasetParams, tree_params: DecisionTreeParams, ) -> ClassificationResult: + dataset: Dataset = Dataset(path=path, file=file) try: - dataset: Dataset = Dataset(path=path, file=file) data = dataset.read(dataset_params) splitted_dataset: SplittedDataset = dataset.split_classification( data=data,