Daryl Fung commited on
Commit
fda0668
1 Parent(s): 9e61b41

fix error ploT

Browse files
Files changed (2) hide show
  1. keyphrase_extraction.py +1 -1
  2. keyword_extraction.py +1 -1
keyphrase_extraction.py CHANGED
@@ -35,7 +35,7 @@ def get_top_key_phrases(text, top_n, save_output):
35
  keyword_df = pd.DataFrame({'words': phrases[:top_n], 'scores': scores[:top_n]})
36
  plt.title("Word Count")
37
  plt.figure(figsize=(24, 8))
38
- sns.barplot(data=keyword_df, y='words', x='scores', palette='blend:#7AB,#EDA', aspect=1.5)
39
  plt.xticks(rotation=-10, fontsize=6)
40
  plt.savefig(save_output, dpi=300)
41
  plt.close()
 
35
  keyword_df = pd.DataFrame({'words': phrases[:top_n], 'scores': scores[:top_n]})
36
  plt.title("Word Count")
37
  plt.figure(figsize=(24, 8))
38
+ sns.barplot(data=keyword_df, y='words', x='scores', palette='blend:#7AB,#EDA')
39
  plt.xticks(rotation=-10, fontsize=6)
40
  plt.savefig(save_output, dpi=300)
41
  plt.close()
keyword_extraction.py CHANGED
@@ -38,7 +38,7 @@ def keyword_extract(doc, kw_model, n_grams, save_output='results/'):
38
  keyword_df = pd.DataFrame({'words': words, 'scores': scores})
39
  plt.title("Word Count")
40
  plt.figure(figsize=(24, 8))
41
- sns.barplot(data=keyword_df, y='words', x='scores', palette='blend:#7AB,#EDA', aspect=1.5)
42
  plt.xticks(rotation=-10, fontsize=8-n_grams)
43
  plt.savefig(save_output, dpi=300)
44
  plt.close()
 
38
  keyword_df = pd.DataFrame({'words': words, 'scores': scores})
39
  plt.title("Word Count")
40
  plt.figure(figsize=(24, 8))
41
+ sns.barplot(data=keyword_df, y='words', x='scores', palette='blend:#7AB,#EDA')
42
  plt.xticks(rotation=-10, fontsize=8-n_grams)
43
  plt.savefig(save_output, dpi=300)
44
  plt.close()