Commit 7a558551 authored by Administrator's avatar Administrator

[FEATURE] Graph increasing edge weight.

parent e2a49e88
......@@ -46,10 +46,6 @@ def create_whitelist(user, corpus):
node_node_ngram AS nngX ON nngX.node_id = n.id
INNER JOIN
node_ngram AS ngX ON ngX.id = nngX.ngram_id
LEFT JOIN
node_ngram AS black ON ngX.terms = black.terms
WHERE
black.terms IS NULL
WHERE
n.parent_id = %d
AND
......@@ -73,7 +69,7 @@ def create_whitelist(user, corpus):
return white_list
#def create_cooc(user, corpus, whitelist, blacklist, synonymes):
def create_cooc(user=None, corpus=None, whitelist=None, size=200):
def create_cooc(user=None, corpus=None, whitelist=None, size=400):
cursor = connection.cursor()
try:
......@@ -201,7 +197,7 @@ def get_cooc(request=None, corpus_id=None, cooc_id=None, type="node_link"):
G.node[node]['size'] = weight[node]
G.node[node]['group'] = partition[node]
# G.node[node]['color'] = '19,180,300'
G.add_edge(node, partition[node], weight=2)
G.add_edge(node, partition[node], weight=3)
except Exception as error:
print(error)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment