minor fixes
This commit is contained in:
parent
a8c10536de
commit
4ea201f410
@ -641,7 +641,7 @@ def main():
|
|||||||
contexts_text.append(line.strip())
|
contexts_text.append(line.strip())
|
||||||
contexts = graph_compiler.texts_to_ids(contexts_text)
|
contexts = graph_compiler.texts_to_ids(contexts_text)
|
||||||
context_graph = ContextGraph(params.context_score)
|
context_graph = ContextGraph(params.context_score)
|
||||||
context_graph.build(contexts)
|
context_graph.build([(c, 0.0) for c in contexts])
|
||||||
else:
|
else:
|
||||||
context_graph = None
|
context_graph = None
|
||||||
else:
|
else:
|
||||||
|
|||||||
@ -686,7 +686,7 @@ def main():
|
|||||||
contexts_text.append(line.strip())
|
contexts_text.append(line.strip())
|
||||||
contexts = graph_compiler.texts_to_ids(contexts_text)
|
contexts = graph_compiler.texts_to_ids(contexts_text)
|
||||||
context_graph = ContextGraph(params.context_score)
|
context_graph = ContextGraph(params.context_score)
|
||||||
context_graph.build(contexts)
|
context_graph.build([(c, 0.0) for c in contexts])
|
||||||
else:
|
else:
|
||||||
context_graph = None
|
context_graph = None
|
||||||
else:
|
else:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user