Commit 955640cd authored by Jim Hoekstra's avatar Jim Hoekstra 👋🏻
Browse files

Merge branch 'issue/MSX-39' into 'develop'

issue MSX-39

See merge request !14
parents 776190ca 66817694
......@@ -85,6 +85,7 @@ class Graph:
elements=elements,
userZoomingEnabled=False,
userPanningEnabled=False,
maxZoom=2.0,
stylesheet=[
{
'selector': '[is_base_node > 0.5]',
......
import os
from dash_app.app import app
from dash_app.layout import layout
import dash_app.callbacks
app.layout = layout
......
......@@ -6,12 +6,12 @@ class AssociatedWords:
def __init__(self):
print("\n Word2Vec model is loading. This can take a couple of minutes.")
# self.model = api.load('glove-twitter-200')
# print(" Word2Vec model is ready. Enjoy!!!\n")
self.model = api.load('glove-twitter-200')
print(" Word2Vec model is ready. Enjoy!!!\n")
def get_associated_words(self, word, top_n=10):
# gensim_result = self.model.most_similar(word, topn=top_n)
gensim_result = [('apple', 1.0), ('banana', 1.0), ('strawberry', 1.0)]
gensim_result = self.model.most_similar(word, topn=top_n)
# gensim_result = [('apple', 1.0), ('banana', 1.0), ('strawberry', 1.0)]
words = self.filter_results(gensim_result, word)
return words
......
Supports Markdown
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