Spaces:
Configuration error
Configuration error
Update evaluate_train_rankings.py
Browse files
evaluate_train_rankings.py
CHANGED
@@ -25,7 +25,7 @@ def main():
|
|
25 |
parser.add_argument('--k_values', type=str, default='3,5,10,20',
|
26 |
help='Comma-separated list of k values for Recall@k')
|
27 |
parser.add_argument('--base_dir', type=str,
|
28 |
-
default=f'{base_directory}/datasets',
|
29 |
help='Base directory for data files')
|
30 |
args = parser.parse_args()
|
31 |
|
|
|
25 |
parser.add_argument('--k_values', type=str, default='3,5,10,20',
|
26 |
help='Comma-separated list of k values for Recall@k')
|
27 |
parser.add_argument('--base_dir', type=str,
|
28 |
+
default=f'{base_directory}/Patent_Retrieval/datasets',
|
29 |
help='Base directory for data files')
|
30 |
args = parser.parse_args()
|
31 |
|