Commit 531b5907 authored by Jim Hoekstra's avatar Jim Hoekstra 👋🏻
Browse files

merge

parent 82674525
......@@ -41,6 +41,7 @@ def update_base_word(submit_word_button, base_word_input):
State(component_id='graph-elements-div', component_property='children'),
State(component_id='msx-graph', component_property='tapNodeData'),
State(component_id='base-word-div', component_property='children'),
# State(component_id='msx-graph', component_property='zoom'),
prevent_initial_call=True
)
def update_graph_elements(submit_word_button, add_word_button, extend_graph_button, remove_word_button, base_word_input,
......@@ -50,6 +51,7 @@ def update_graph_elements(submit_word_button, add_word_button, extend_graph_butt
button_id = callback_context.triggered[0]['prop_id'].split('.')[0]
graph = Graph()
# print('graph zoom level: ', graph_zoom_level)
if button_id == 'submit-word-button':
graph.fill_with_associations(word2vec_model, base_word_input)
......
import os
from dash_app.app import app
from dash_app.layout import layout
import dash_app.callbacks
app.layout = layout
......
......@@ -5,13 +5,13 @@ import stringdist as sdi
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")
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")
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