diff --git a/Train.py b/Train.py index b586dbcdacad70bff17f20d4fe7d7ed2b34d9799..11215ed8665405de21bab5ff4314697a59780baf 100644 --- a/Train.py +++ b/Train.py @@ -172,7 +172,7 @@ def trainModelRl(debug, modelDir, filename, nbIter, batchSize, devFile, transiti missingLinks = getMissingLinks(sentence) if debug : sentence.printForDebug(sys.stderr) - action = selectAction(policy_net, state, transitionSet, sentence, missingLinks, probaRandom=0.3, probaOracle=0.15) + action = selectAction(policy_net, state, transitionSet, sentence, missingLinks, probaRandom=0.1, probaOracle=0.1) if action is None : break