diff --git a/main.py b/main.py index cff4887..49d450d 100644 --- a/main.py +++ b/main.py @@ -112,8 +112,8 @@ if __name__ == '__main__': parser.add_argument('dataset', help='Path to the dataset') parser.add_argument('-o', '--output', dest='csv_dir', metavar='', - help='Result file (default ../csv_logs/gfun/gfun_results.csv)', type=str, - default='../csv_logs/gfun/gfun_results.csv') + help='Result file (default csv_logs/gfun/gfun_results.csv)', type=str, + default='csv_logs/gfun/gfun_results.csv') parser.add_argument('-x', '--post_embedder', dest='post_embedder', action='store_true', help='deploy posterior probabilities embedder to compute document embeddings', diff --git a/src/util/common.py b/src/util/common.py index 913014c..9f44273 100644 --- a/src/util/common.py +++ b/src/util/common.py @@ -378,7 +378,7 @@ def get_method_name(args): for i, conf in enumerate(_id_conf): if conf: _id += _id_name[i] - _id = _id if not args.gru_wce else _id + '_wce' + _id = _id if not args.rnn_wce else _id + '_wce' _dataset_path = args.dataset.split('/')[-1].split('_') dataset_id = _dataset_path[0] + _dataset_path[-1] return _id, dataset_id