Fix dataset error
This commit is contained in:
parent
dc3d3620b7
commit
5817314562
@ -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,
|
||||
|
Loading…
x
Reference in New Issue
Block a user