diff --git a/core/classifiers.py b/core/classifiers.py index db9fdce..0416edc 100644 --- a/core/classifiers.py +++ b/core/classifiers.py @@ -214,8 +214,8 @@ def predict_subclasses(self, text, n=5, limit_to=None): self._load_in_memory() x = self._to_feature_vector(text) - y = self.model.predict_step(x)[0] - + y = self.model.predict(x)[0] + # sort subclasses in descending order of relevancy subclasses = [self.subclass_codes[i] for i in np.argsort(y)[::-1]]