From 94eb5badd3c52d7f4f192e3c6b6cdb6f05069e80 Mon Sep 17 00:00:00 2001 From: kevcruse96 Date: Wed, 3 Nov 2021 15:39:36 -0700 Subject: [PATCH] Update README --- README.md | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/README.md b/README.md index 86b80f2..c6f47a1 100644 --- a/README.md +++ b/README.md @@ -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), @@ -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) ``` \ No newline at end of file