andrea.pedrotti
/
gFun
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed arguments

This commit is contained in:
andrea 2021-02-02 11:26:04 +01:00
parent bca0b9ab7c
commit 1bff57ebbb
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

@ -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',

2
src/util/common.py
View File

@ -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