Skip to content
Snippets Groups Projects
Commit d9fc23a0 authored by Franck Dary's avatar Franck Dary
Browse files

network provide the column name it needs

parent 44403c5a
Branches
No related tags found
No related merge requests found
......@@ -210,7 +210,7 @@ def trainModelRl(debug, networkName, modelDir, filename, nbIter, batchSize, devF
if os.path.isfile(modelDir+"/dicts.json") :
dicts.load(modelDir+"/dicts.json")
else :
dicts.readConllu(filename, ["FORM","UPOS","LETTER","LEXICON"], 2, pretrained)
dicts.readConllu(filename, Networks.getNeededDicts(networkName), 2, pretrained)
dicts.addDict("HISTORY", transitionNames)
dicts.save(modelDir + "/dicts.json")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment