Skip to content

Commit

Permalink
Update README
Browse files Browse the repository at this point in the history
  • Loading branch information
kevcruse96 committed Nov 3, 2021
1 parent b228977 commit 94eb5ba
Showing 1 changed file with 4 additions and 8 deletions.
12 changes: 4 additions & 8 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -78,9 +78,9 @@ from synthesis_action_retriever.synthesis_action_retriever import SynthActionRet
from synthesis_action_retriever.build_graph import GraphBuilder
from synthesis_action_retriever.utils import make_spacy_tokens
dir_path = "/Users/kevcruse96/Desktop/D2S2/Saved_Models/sar_models"
w2v_model = "w2v_embeddings_v2_words_420K"
ext_model = "Bi-RNN_cl7_ed100_TF_20211018-122820"
dir_path = "path-to-models"
w2v_model = "path-to-w2v_model"
ext_model = "path-to-ext_model"
sar = SynthActionRetriever(
embedding_model=os.path.join(dir_path, w2v_model),
Expand All @@ -97,10 +97,6 @@ for sent in examples:
actions = sar.get_action_labels(spacy_tokens)
graph.append(gb.build_graph(spacy_tokens, actions, sent["materials"]))
para = ' '.join([s["sentence"] for s in examples])
para_sent_toks = Paragraph(para).raw_tokens
refined_graph = gb.refine_graph(graph, examples, para_sent_toks)
refined_graph = gb.refine_graph(graph, examples)
pprint(refined_graph)
```

0 comments on commit 94eb5ba

Please sign in to comment.