Commit bd24b8a3 authored by delanoe's avatar delanoe

[FIX] Adding last files for previous commit.

parent 20061ba9
...@@ -137,24 +137,20 @@ def compute_groups(corpus, limit_inf=None, limit_sup=None, how='Stem'): ...@@ -137,24 +137,20 @@ def compute_groups(corpus, limit_inf=None, limit_sup=None, how='Stem'):
miam_to_insert.add((miam_node.id, n[0],1)) miam_to_insert.add((miam_node.id, n[0],1))
#print([n for n in group]) #print([n for n in group])
for g in group: for g in group:
if (miam_node.id, g[0],1) not in miam_to_insert: if (miam_node.id, g[0], 1) not in miam_to_insert:
#list_to_check.remove(g)
group_to_insert.append((node_group.id, n[0], g[0], 1)) group_to_insert.append((node_group.id, n[0], g[0], 1))
print(n[1], "=", g[1]) print(n[1], "=", g[1])
# TODO see here if coherent add in miam or group...
# Deleting previous groups # Deleting previous groups
session.query(NodeNgramNgram).filter(NodeNgramNgram.node_id == node_group.id).delete() session.query(NodeNgramNgram).filter(NodeNgramNgram.node_id == node_group.id).delete()
bulk_insert(NodeNgramNgram bulk_insert(NodeNgramNgram
, ('node_id', 'ngramx_id', 'ngramy_id', 'score') , ('node_id', 'ngramx_id', 'ngramy_id', 'score')
, [data for data in group_to_insert]) , [data for data in group_to_insert])
# # stop_id = get_or_create_node(parent_id=corpus.id,user_id=corpus.user_id,nodetype='StopList')
#session.commit()
for n in group_to_insert: for n in group_to_insert:
print(n) #print(n)
miam_to_insert.add((miam_node.id, n[1], 1)) miam_to_insert.add((miam_node.id, n[1], 1))
# Deleting previous ngrams miam list # Deleting previous ngrams miam list
session.query(NodeNgram).filter(NodeNgram.node_id == miam_node.id).delete() session.query(NodeNgram).filter(NodeNgram.node_id == miam_node.id).delete()
bulk_insert(NodeNgram, ('node_id', 'ngram_id', 'weight'), [data for data in list(miam_to_insert)]) bulk_insert(NodeNgram, ('node_id', 'ngram_id', 'weight'), [data for data in list(miam_to_insert)])
#
...@@ -27,7 +27,7 @@ def listIds(typeList=None, user_id=None, corpus_id=None): ...@@ -27,7 +27,7 @@ def listIds(typeList=None, user_id=None, corpus_id=None):
# Nodes are either in root_list or user_list # Nodes are either in root_list or user_list
root_list = ['Stem', 'Lem'] root_list = ['Stem', 'Lem']
user_list = ['MiamList', 'StopList', 'MainList', 'GroupList'] user_list = ['MiamList', 'StopList', 'MapList', 'Group']
if typeList in user_list: if typeList in user_list:
nodes = session.query(Node).filter( nodes = session.query(Node).filter(
......
...@@ -364,15 +364,15 @@ class NodeNgramNgram(models.Model): ...@@ -364,15 +364,15 @@ class NodeNgramNgram(models.Model):
return "%s: %s / %s" % (self.node.name, self.ngramx.terms, self.ngramy.terms) return "%s: %s / %s" % (self.node.name, self.ngramx.terms, self.ngramy.terms)
class NodeHyperdataNgram(models.Model): class NodeHyperdataNgram(models.Model):
node = models.ForeignKey(Node, on_delete=models.CASCADE) node = models.ForeignKey(Node, on_delete=models.CASCADE)
hyperdata_id = models.ForeignKey(Hyperdata, related_name="nodehyperdatamngramx", on_delete=models.CASCADE) hyperdata = models.ForeignKey(Hyperdata, on_delete=models.CASCADE)
ngram_id = models.ForeignKey(Ngram, related_name="nodehyperdatangramy", on_delete=models.CASCADE) ngram = models.ForeignKey(Ngram, on_delete=models.CASCADE)
score = models.FloatField(default=0) score = models.FloatField(default=0)
def __str__(self): def __str__(self):
return "%s: %s / %s" % (self.node.name, self.hyperdata_id.value_string, self.ngram_id.terms) return "%s: %s / %s" % (self.node.name, self.hyperdata.value_string, self.ngram.terms)
class NodeNodeNgram(models.Model): class NodeNodeNgram(models.Model):
......
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