fix bar plot

keyphrase_extraction.py

@@ -35,9 +35,9 @@ def get_top_key_phrases(text, top_n, save_output):
     keyword_df = pd.DataFrame({'words': phrases[:top_n], 'scores': scores[:top_n]})
     plt.title("Word Count")
     plt.figure(figsize=(24, 8))
-    plt.yticks(fontsize=12)
+    plt.yticks(fontsize=15)
     sns.barplot(data=keyword_df, y='words', x='scores', palette='blend:#7AB,#EDA')
-    plt.savefig(save_output, dpi=300)
+    plt.savefig(save_output, dpi=300, bbox_inches="tight")
     plt.close()
 
 

keyword_extraction.py

@@ -38,9 +38,9 @@ def keyword_extract(doc, kw_model, n_grams, save_output='results/'):
     keyword_df = pd.DataFrame({'words': words, 'scores': scores})
     plt.title("Word Count")
     plt.figure(figsize=(24, 8))
-    plt.yticks(fontsize=12)
+    plt.yticks(fontsize=15)
     sns.barplot(data=keyword_df, y='words', x='scores', palette='blend:#7AB,#EDA')
-    plt.savefig(save_output, dpi=300)
+    plt.savefig(save_output, dpi=300, bbox_inches="tight")
     plt.close()
 
 if __name__ == '__main__':

word.py

@@ -34,9 +34,9 @@ def show_gram_plot(doc, n_grams, top_words=10, save_output='results/'):
     tokens_df = pd.DataFrame({'word': words, 'count': count})
     plt.title("Word Count")
     plt.figure(figsize=(24, 8))
-    plt.yticks(fontsize=12)
+    plt.yticks(fontsize=15)
     sns.barplot(data=tokens_df, y='word', x='count', palette='blend:#7AB,#EDA')
-    plt.savefig(save_output, dpi=300)
+    plt.savefig(save_output, dpi=300, bbox_inches="tight")
     plt.close()