This commit is contained in:
Andrea Pedrotti 2024-03-12 11:40:02 +01:00
parent 35cc32e541
commit 59bf921bf3
2 changed files with 2 additions and 3 deletions

View File

@ -57,7 +57,6 @@ class SimpleGfunDataset:
print(f"tr: {tr} - va: {va} - te: {te}")
def load_csv_inference(self):
# test = pd.read_csv(os.path.join(self.datadir, "test.small.csv" if not self.reduced else "test.small.csv"))
test = pd.read_csv(self.datadir)
self._set_labels(test)
self._set_langs(train=None, test=test)

View File

@ -14,10 +14,10 @@ def load_from_pickle(path, dataset_name, nrows):
return loaded
def get_dataset(datasetp_path, args):
def get_dataset(dataset_path, args):
dataset = SimpleGfunDataset(
dataset_name="rai",
datapath=datasetp_path,
datapath=dataset_path,
textual=True,
visual=False,
multilabel=False,