update info

absa_evaluator.py

@@ -12,29 +12,33 @@ _CITATION = """
 """
 
 _DESCRIPTION = """
-Evaluation metrics for Aspect-Based Sentiment Analysis (ABSA) including precision, recall, and F1 score for aspect terms and polarities.
+This module provides evaluation metrics for Aspect-Based Sentiment Analysis (ABSA). 
+The metrics include precision, recall, and F1 score for both aspect terms and category detection.
+Additionally it calculates de accuracy for polarities from aspect terms and category detection.
+ABSA evaluates the capability of a model to identify and correctly classify the sentiment of specific aspects within a text.
 """
 
 _KWARGS_DESCRIPTION = """
-Computes precision, recall, and F1 score for aspect terms and polarities in Aspect-Based Sentiment Analysis (ABSA).
+Computes precision, recall, and F1 score for aspect terms and category detection in Aspect-Based Sentiment Analysis (ABSA). Also calculates de accuracy for polarities on each task.
 
 Args:
     predictions: List of ABSA predictions with the following structure:
         - 'aspects': Sequence of aspect annotations, each with the following keys:
             - 'term': Aspect term
             - 'polarity': Polarity of the aspect term
+        - 'category': Sequence of category annotations, each with the following keys:
+            - 'category': Category
+            - 'polarity': polarity of the category
     references: List of ABSA references with the same structure as predictions.
 Returns:
-    aspect_precision: Precision score for aspect terms
-    aspect_recall: Recall score for aspect terms
-    aspect_f1: F1 score for aspect terms
-    polarity_precision: Precision score for aspect polarities
-    polarity_recall: Recall score for aspect polarities
-    polarity_f1: F1 score for aspect polarities
+    term_extraction_results: f1 score, precision and recall for aspect terms
+    term_polarity_results_accuracy: accuracy for polarities on aspect terms
+    category_detection_results: f1 score, precision and recall for category detection
+    category_polarity_results_accuracy: accuracy for polarities on categories
 """
 
 
-class AbsaEvaluatorTest(evaluate.Metric):
+class AbsaEvaluator(evaluate.Metric):
     def _info(self):
         return evaluate.MetricInfo(
             description=_DESCRIPTION,