Update modules/semantic_analysis.py
Browse files
modules/semantic_analysis.py
CHANGED
@@ -85,6 +85,9 @@ POS_TRANSLATIONS = {
|
|
85 |
}
|
86 |
########################################################################################################################################
|
87 |
|
|
|
|
|
|
|
88 |
def extract_entities(doc):
|
89 |
entities = {
|
90 |
"Personas": [],
|
@@ -154,10 +157,10 @@ def create_semantic_graph(doc, lang):
|
|
154 |
if token.dep_ != "ROOT":
|
155 |
G.add_edge(token.head.text, token.text, label=token.dep_)
|
156 |
|
157 |
-
return G
|
158 |
|
159 |
def visualize_semantic_relations(doc, lang):
|
160 |
-
G = create_semantic_graph(doc, lang)
|
161 |
|
162 |
plt.figure(figsize=(24, 18))
|
163 |
pos = nx.spring_layout(G, k=0.9, iterations=50)
|
|
|
85 |
}
|
86 |
########################################################################################################################################
|
87 |
|
88 |
+
def count_pos(doc):
|
89 |
+
return Counter(token.pos_ for token in doc if token.pos_ != 'PUNCT')
|
90 |
+
|
91 |
def extract_entities(doc):
|
92 |
entities = {
|
93 |
"Personas": [],
|
|
|
157 |
if token.dep_ != "ROOT":
|
158 |
G.add_edge(token.head.text, token.text, label=token.dep_)
|
159 |
|
160 |
+
return G, pos_counts
|
161 |
|
162 |
def visualize_semantic_relations(doc, lang):
|
163 |
+
G, pos_counts = create_semantic_graph(doc, lang)
|
164 |
|
165 |
plt.figure(figsize=(24, 18))
|
166 |
pos = nx.spring_layout(G, k=0.9, iterations=50)
|