diff --git a/unimore_bda_6/analysis/tf_text.py b/unimore_bda_6/analysis/tf_text.py index 234b786..260429d 100644 --- a/unimore_bda_6/analysis/tf_text.py +++ b/unimore_bda_6/analysis/tf_text.py @@ -260,7 +260,7 @@ class TensorflowPolarSentimentAnalyzer(TensorflowSentimentAnalyzer): return model def _translate_prediction(self, a: numpy.array) -> Category: - return a[0, 0] + return a[0, 0] * 5 __all__ = (