update

yangheng95 · Aug 22, 2022 · f29e511 · f29e511
1 parent ffd1ff0
commit f29e511
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/pyabsa/core/apc/classic/__glove__/dataset_utils/data_utils_for_inference.py b/pyabsa/core/apc/classic/__glove__/dataset_utils/data_utils_for_inference.py
@@ -67,6 +67,8 @@ def __init__(self, tokenizer, opt):
         self.all_data = []
 
     def parse_sample(self, text):
+        if '[ASP]' not in text:
+            text = '[ASP] Global Sentiment [ASP]' + text
         _text = text
         samples = []
 

diff --git a/pyabsa/core/apc/prediction/sentiment_classifier.py b/pyabsa/core/apc/prediction/sentiment_classifier.py
@@ -206,8 +206,6 @@ def infer(self, text: str = None,
               print_result=True,
               ignore_error=True,
               clear_input_samples=True):
-        if text.count('[ASP]') < 2:
-            text = '[ASP] Global Sentiment [ASP]' + text
         if clear_input_samples:
             self.clear_input_samples()
         if text:

diff --git a/setup.py b/setup.py
@@ -32,7 +32,7 @@
     license='MIT',
     install_requires=['findfile>=1.7.9.8',
                       'autocuda>=0.11',
-                      'metric-visualizer>=0.5.4',
+                      'metric-visualizer>=0.5.5',
                       'boostaug>=2.2.3',
                       'spacy',
                       'networkx',