diff --git a/opentapioca/classifier.py b/opentapioca/classifier.py index c6c4fc1..da0aff2 100644 --- 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)