update

shuyuan-lily · Oct 29, 2022 · 5ada6d2 · 5ada6d2
1 parent 7991f19
commit 5ada6d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/demos/aspect_term_extraction/extract_aspects.py b/demos/aspect_term_extraction/extract_aspects.py
@@ -22,4 +22,4 @@
                                                pred_sentiment=True,  # Predict the sentiment of extracted aspect terms
                                                )
 
-# print(atepc_result)
+print(atepc_result)
diff --git a/pyabsa/core/apc/classic/__bert__/dataset_utils/data_utils_for_training.py b/pyabsa/core/apc/classic/__bert__/dataset_utils/data_utils_for_training.py
@@ -103,7 +103,7 @@ def __init__(self, dataset_list, tokenizer, opt):
             if lines[i].count("$T$") > 1:
                 continue
             text_left, _, text_right = [s.lower().strip() for s in lines[i].partition("$T$")]
-            aspect = lines[i + 1].strip()
+            aspect = lines[i + 1].lower().strip()
             text_raw = text_left + ' ' + aspect + ' ' + text_right
             polarity = lines[i + 2].strip()
             # polarity = int(polarity)