1
0
Fork 0

pathfix in lequa2022 datasets

This commit is contained in:
Alejandro Moreo Fernandez 2022-06-15 14:36:02 +02:00
parent 2cc7db60cc
commit 789b9d5fbc
1 changed files with 1 additions and 1 deletions

View File

@ -580,7 +580,7 @@ def fetch_lequa2022(task, data_home=None):
test_samples_path = join(lequa_dir, task, 'public', 'dev_samples')
test_true_prev_path = join(lequa_dir, task, 'public', 'test_prevalences.txt')
test_gen = SamplesFromDir(val_samples_path, val_true_prev_path, load_fn=load_fn)
test_gen = SamplesFromDir(test_samples_path, test_true_prev_path, load_fn=load_fn)
return train, val_gen, test_gen