Fix dataset error

This commit is contained in:
Aleksey Filippov 2025-03-11 23:35:23 +04:00
parent dc3d3620b7
commit 5817314562

View File

@ -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,