andrea.pedrotti
/
gfun_multimodal
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

typos

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
dataManager/gFunDataset.py
View File

@ -57,7 +57,6 @@ class SimpleGfunDataset:
print(f"tr: {tr} - va: {va} - te: {te}") print(f"tr: {tr} - va: {va} - te: {te}")
def load_csv_inference(self): 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) test = pd.read_csv(self.datadir)
self._set_labels(test) self._set_labels(test)
self._set_langs(train=None, test=test) self._set_langs(train=None, test=test)

4
dataManager/utils.py
View File

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