adding name to datasets un fetch_UCIMulticlassDataset
This commit is contained in:
parent
a6ff00f96b
commit
55c62a9dd2
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue