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

running comparison with refactor branch

This commit is contained in:
andrea 2021-01-29 11:37:42 +01:00
parent 5906f85f33
commit 091101b39d
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/main_gFun.py
View File

@ -87,8 +87,9 @@ if __name__ == '__main__':
NB: --allprob won't have any effect on this View Gen since output is already encoded as post prob NB: --allprob won't have any effect on this View Gen since output is already encoded as post prob
""" """
op.gru_path = '/home/andreapdr/funneling_pdr/checkpoint/gru_viewgen_-rcv1-2_doclist_trByLang1000_teByLang1000_processed_run0.pickle' # TODO DEBUG op.gru_path = '/home/andreapdr/funneling_pdr/checkpoint/gru_viewgen_-rcv1-2_doclist_trByLang1000_teByLang1000_processed_run0.pickle' # TODO DEBUG
op.gru_path = None
rnn_embedder = RecurrentEmbedder(pretrained=op.gruMUSE, supervised=op.gruWCE, multilingual_dataset=data, rnn_embedder = RecurrentEmbedder(pretrained=op.gruMUSE, supervised=op.gruWCE, multilingual_dataset=data,
options=op, model_path=op.gru_path) options=op, model_path=op.gru_path, we_path=op.we_path)
doc_embedder.append(rnn_embedder) doc_embedder.append(rnn_embedder)
if op.mbert: if op.mbert: