Merge pull request #14 from heathersherry/patch-1

Update classifier.py
opentapioca · Dec 10, 2019 · 10890f9 · 10890f9
2 parents bc6ebb2 + 25f2c68
commit 10890f9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/opentapioca/classifier.py b/opentapioca/classifier.py
@@ -295,6 +295,8 @@ def build_feature_vectors_for_doc(self, mentions):
 
         mixed_features = feature_array
         mixed_features_array = [feature_array]
+        mixed_features = mixed_features.astype(float)
+
         for i in range(self.nb_steps):
             mixed_features = numpy.dot(adj_matrix, mixed_features)
             mixed_features_array.append(mixed_features)