Minor bugfix.

Bartłomiej Nitoń
1 parent 3fa58087
Showing 1 changed file with 3 additions and 14 deletions
corneferencer/resolvers/resolve.py
@@ -123,20 +123,9 @@ def closest(text, threshold):
         if i > 0:
             for ante in reversed(text.mentions[:i]):
                 if not features.pair_intersect(ante, ana):
-                    pair_features = get_pair_features(ante, ana)
-
-                    ante_vec = []
-                    ante_vec.extend(ante.features)
-                    ante_vec.extend(pair_features)
-                    ante_sample = numpy.asarray([ante_vec], dtype=numpy.float32)
-
-                    ana_vec = []
-                    ana_vec.extend(ana.features)
-                    ana_vec.extend(pair_features)
-                    ana_sample = numpy.asarray([ana_vec], dtype=numpy.float32)
-
-                    prediction = NEURAL_MODEL.predict([ante_sample, ana_sample])[0]
-
+                    pair_vec = get_pair_vector(ante, ana)
+                    sample = numpy.asarray([pair_vec], dtype=numpy.float32)
+                    prediction = NEURAL_MODEL.predict(sample)[0]
                     if prediction > threshold:
                         if ante.set:
                             ana.set = ante.set