From b8acf5fc7cad1be009f87b58670cc9a5277101ea Mon Sep 17 00:00:00 2001 From: Stefano Pigozzi Date: Thu, 2 Feb 2023 04:07:28 +0100 Subject: [PATCH] Group documents in just two categories --- unimore_bda_6/analysis/vanilla.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/unimore_bda_6/analysis/vanilla.py b/unimore_bda_6/analysis/vanilla.py index 249dedd..9b3406b 100644 --- a/unimore_bda_6/analysis/vanilla.py +++ b/unimore_bda_6/analysis/vanilla.py @@ -83,14 +83,11 @@ class VanillaReviewSA(VanillaSA): Return the label corresponding to the given rating. Possible categories are: - * negative (0.0 <= rating < 2.5) - * mixed (2.5 <= rating <= 3.5) - * positive (3.5 < rating <= 5.0) + * negative (0.0 <= rating < 3.0) + * positive (3.0 < rating <= 5.0) """ - if rating < 2.5: + if rating < 3.0: return "negative" - elif rating <= 3.5: - return "mixed" else: return "positive"