Update subject_extract.py

hiworldwzj · Jun 18, 2019 · 6bc8e05 · 6bc8e05
1 parent 93c9b00
commit 6bc8e05
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/subject_extract.py b/subject_extract.py
@@ -251,7 +251,7 @@ def evaluate(self):
 def test(test_data):
  F = open('result.txt', 'w')
  for d in tqdm(iter(test_data)):
- s = u'"%s","%s"\n' % (d[0], extract_entity(d[1].replace('\t', ''), d[2]))
+ s = u'"%s","%s"\n' % (d[0], extract_entity(d[1], d[2]))
  s = s.encode('utf-8')
  F.write(s)
  F.close()
@@ -264,7 +264,7 @@ def test(test_data):
 if __name__ == '__main__':
  train_model.fit_generator(train_D.__iter__(),
  steps_per_epoch=len(train_D),
- epochs=20,
+ epochs=120,
  callbacks=[evaluator]
  )
 else: