AIdeaText commited on
Commit
bd04299
1 Parent(s): 471b1ae

Update modules/syntax_analysis.py

Browse files
Files changed (1) hide show
  1. modules/syntax_analysis.py +4 -4
modules/syntax_analysis.py CHANGED
@@ -83,10 +83,10 @@ POS_TRANSLATIONS = {
83
  'X': 'Autre',
84
  }
85
  }
86
-
87
  def count_pos(doc):
88
  return Counter(token.pos_ for token in doc if token.pos_ != 'PUNCT')
89
-
90
  def create_syntax_graph(doc, lang):
91
  G = nx.DiGraph()
92
  pos_counts = count_pos(doc)
@@ -113,7 +113,7 @@ def create_syntax_graph(doc, lang):
113
  G.add_edge(head_id, word_nodes[lower_text], label=token.dep_)
114
 
115
  return G, word_colors
116
-
117
  def visualize_syntax_graph(doc, lang):
118
  G, word_colors = create_syntax_graph(doc, lang)
119
 
@@ -142,7 +142,7 @@ def visualize_syntax_graph(doc, lang):
142
  plt.legend(handles=legend_elements, loc='center left', bbox_to_anchor=(1, 0.5), fontsize=12) # Increase legend font size
143
 
144
  return plt
145
-
146
  def visualize_syntax(text, nlp, lang):
147
  max_tokens = 5000
148
  doc = nlp(text)
 
83
  'X': 'Autre',
84
  }
85
  }
86
+ ########################################################################################################################################
87
  def count_pos(doc):
88
  return Counter(token.pos_ for token in doc if token.pos_ != 'PUNCT')
89
+ #######################################################################################################################################
90
  def create_syntax_graph(doc, lang):
91
  G = nx.DiGraph()
92
  pos_counts = count_pos(doc)
 
113
  G.add_edge(head_id, word_nodes[lower_text], label=token.dep_)
114
 
115
  return G, word_colors
116
+ ####################################################################################################################################
117
  def visualize_syntax_graph(doc, lang):
118
  G, word_colors = create_syntax_graph(doc, lang)
119
 
 
142
  plt.legend(handles=legend_elements, loc='center left', bbox_to_anchor=(1, 0.5), fontsize=12) # Increase legend font size
143
 
144
  return plt
145
+ ################################################################################################################################
146
  def visualize_syntax(text, nlp, lang):
147
  max_tokens = 5000
148
  doc = nlp(text)