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:, :]