From 46c24d9fd85be6e35b4c290d28f606dd3d59a2b0 Mon Sep 17 00:00:00 2001 From: Lorenzo Volpi Date: Sat, 3 Feb 2024 12:38:29 +0100 Subject: [PATCH] dataset fixed --- quacc/dataset.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/quacc/dataset.py b/quacc/dataset.py index 9232308..37f8f8d 100644 --- a/quacc/dataset.py +++ b/quacc/dataset.py @@ -126,9 +126,7 @@ class DatasetProvider: # provare min_df=5 def __imdb(self, **kwargs): - return qp.datasets.fetch_reviews( - "imdb", data_home="./quapy_data", tfidf=True, min_df=3 - ).train_test + return qp.datasets.fetch_reviews("imdb", tfidf=True, min_df=3).train_test def __rcv1(self, target, **kwargs): n_train = 23149 @@ -137,7 +135,7 @@ class DatasetProvider: if target is None or target not in available_targets: raise ValueError(f"Invalid target {target}") - dataset = fetch_rcv1(data_home="./scikit_learn_data") + dataset = fetch_rcv1() target_index = np.where(dataset.target_names == target)[0] all_train_d = dataset.data[:n_train, :] test_d = dataset.data[n_train:, :]