From 0a4ce38982afec3a7c0a7e8eb371c886f1004e00 Mon Sep 17 00:00:00 2001 From: Stefano Pigozzi Date: Fri, 10 Feb 2023 04:17:50 +0100 Subject: [PATCH] Use the correct sentiment analyzer in main --- unimore_bda_6/__main__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/unimore_bda_6/__main__.py b/unimore_bda_6/__main__.py index 84d0ec8..587fe84 100644 --- a/unimore_bda_6/__main__.py +++ b/unimore_bda_6/__main__.py @@ -7,7 +7,7 @@ install_general_log_handlers() from .config import config from .database import mongo_client_from_config, reviews_collection, sample_reviews_polar, sample_reviews_varied from .analysis.nltk_sentiment import NLTKSentimentAnalyzer -from .analysis.tf_text import TensorflowSentimentAnalyzer +from .analysis.tf_text import TensorflowCategorySentimentAnalyzer from .analysis.base import TrainingFailedError from .tokenizer import PlainTokenizer, LowercaseTokenizer, NLTKWordTokenizer, PottsTokenizer, PottsTokenizerWithNegation from .gathering import Caches @@ -39,7 +39,7 @@ def main(): slog.debug("Selected sample_func: %s", sample_func.__name__) for SentimentAnalyzer in [ - TensorflowSentimentAnalyzer, + TensorflowCategorySentimentAnalyzer, NLTKSentimentAnalyzer ]: