Skip to content
Snippets Groups Projects

Fix bug with double predictions - develop to master merge.

Merged Mateusz Klimaszewski requested to merge develop into master
Compare and
3 files
+ 27
35
Compare changes
  • Side-by-side
  • Inline
Files
3
+ 3
4
@@ -84,8 +84,8 @@ flags.DEFINE_integer(name="batch_size", default=1,
help="Prediction batch size.")
flags.DEFINE_boolean(name="silent", default=True,
help="Silent prediction to file (without printing to console).")
flags.DEFINE_enum(name="predictor_name", default="semantic-multitask-predictor-spacy",
enum_values=["semantic-multitask-predictor", "semantic-multitask-predictor-spacy"],
flags.DEFINE_enum(name="predictor_name", default="combo-spacy",
enum_values=["combo", "combo-spacy"],
help="Use predictor with whitespace or spacy tokenizer.")
@@ -151,8 +151,7 @@ def run(_):
if FLAGS.input_file == "-":
use_dataset_reader = False
predictor.without_sentence_embedding = True
if use_dataset_reader:
predictor.line_to_conllu = True
predictor.line_to_conllu = False
if FLAGS.silent:
logging.getLogger("allennlp.common.params").disabled = True
manager = allen_predict._PredictManager(