From d9fc23a05382ae04f78a11896b0e1ac88791d16a Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Wed, 18 May 2022 12:12:11 +0200
Subject: [PATCH] network provide the column name it needs
---
Train.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/Train.py b/Train.py
index a15c600..ca19657 100644
--- a/Train.py
+++ b/Train.py
@@ -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")
--
GitLab