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:
|
if n_train > max_train_instances:
|
||||||
train_prop = (max_train_instances / n)
|
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:
|
if standardize:
|
||||||
data = standardizer(data)
|
data = standardizer(data)
|
||||||
|
|
Loading…
Reference in New Issue