using portuguese terms

src/datashap/DataSHAP.py

@@ -83,15 +83,17 @@ class DataSHAP:
 
     def get_statistic(self, df)->DataFrame:
         statistic = DataFrame()
-        statistic['positive_score'] = df[df['tag']=='POSITIVE']['score'].describe()
-        statistic['negative_score'] = df[df['tag']=='NEGATIVE']['score'].describe()
-        statistic['neutral_score']  = df[df['tag']=='NEUTRAL']['score'].describe()
+        statistic['Score Positivo'] = df[df['tag']=='POSITIVE']['score'].describe()
+        statistic['Score Negativo'] = df[df['tag']=='NEGATIVE']['score'].describe()
+        statistic['Score Neutro']  = df[df['tag']=='NEUTRAL']['score'].describe()
         return statistic
     
     def get_performance(self):
         return self.total_tokens, self.h, self.m, self.s
     
     def three_plot(self, df):
+        df['tag'] = df['tag'].replace({'POSITIVE': 'Positivo', 'NEGATIVE': 'Negativo', 'NEUTRAL': 'Neutro'})
+        df = df.rename(columns={'tag': 'rotulo'})
         fig = make_subplots(rows=2, cols=2, horizontal_spacing = 0.0, vertical_spacing = 0.05,
                             shared_xaxes=True, shared_yaxes=True,
                             row_heights=[0.4, 0.6], column_widths=[0.8, 0.2])