From 58b9f364db5176345539591f871b0e0feecdc53a Mon Sep 17 00:00:00 2001 From: Najaf Murtaza <16462407+najafmurtaza@users.noreply.github.com> Date: Sun, 16 Aug 2020 00:59:53 +0500 Subject: [PATCH] Fixed examples/gensen.py ```examples/gensen.py, batcher``` func has undefined ```gensen``` and ```sentences``` variables. Now they are replaced with their appropriate var names. --- examples/gensen.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/examples/gensen.py b/examples/gensen.py index 46dbc5b6..4e1e3bc6 100644 --- a/examples/gensen.py +++ b/examples/gensen.py @@ -31,8 +31,8 @@ def prepare(params, samples): def batcher(params, batch): batch = [' '.join(sent) if sent != [] else '.' for sent in batch] - _, reps_h_t = gensen.get_representation( - sentences, pool='last', return_numpy=True, tokenize=True + _, reps_h_t = params['gensen'].get_representation( + batch, pool='last', return_numpy=True, tokenize=True ) embeddings = reps_h_t return embeddings @@ -49,9 +49,6 @@ def batcher(params, batch): pretrained_emb='../data/embedding/glove.840B.300d.h5' ) gensen_encoder = GenSen(gensen_1, gensen_2) -reps_h, reps_h_t = gensen.get_representation( - sentences, pool='last', return_numpy=True, tokenize=True -) # Set params for SentEval params_senteval = {'task_path': PATH_TO_DATA, 'usepytorch': True, 'kfold': 5}