From 47d7137c837de33207230cb780ab369019f50ff8 Mon Sep 17 00:00:00 2001 From: Kumaran Rajendhiran Date: Wed, 5 Apr 2017 16:48:35 +0530 Subject: [PATCH] Set max_length to 100 for demo and evaluate --- examples/keras_parikh_entailment/spacy_hook.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/keras_parikh_entailment/spacy_hook.py b/examples/keras_parikh_entailment/spacy_hook.py index 0177da001..c1e36327c 100644 --- a/examples/keras_parikh_entailment/spacy_hook.py +++ b/examples/keras_parikh_entailment/spacy_hook.py @@ -80,10 +80,10 @@ def get_word_ids(docs, rnn_encode=False, tree_truncate=False, max_length=100, nr return Xs -def create_similarity_pipeline(nlp): +def create_similarity_pipeline(nlp, max_length=100): return [ nlp.tagger, nlp.entity, nlp.parser, - KerasSimilarityShim.load(nlp.path / 'similarity', nlp, max_length=10) + KerasSimilarityShim.load(nlp.path / 'similarity', nlp, max_length) ]