adding name to datasets un fetch_UCIMulticlassDataset

This commit is contained in:
Alejandro Moreo Fernandez 2024-09-16 13:22:36 +02:00
parent a6ff00f96b
commit 55c62a9dd2
1 changed files with 1 additions and 1 deletions

View File

@ -637,7 +637,7 @@ def fetch_UCIMulticlassDataset(
if n_train > max_train_instances:
train_prop = (max_train_instances / n)
data = Dataset(*data.split_stratified(train_prop, random_state=0))
data = Dataset(*data.split_stratified(train_prop, random_state=0), name=dataset_name)
if standardize:
data = standardizer(data)