def build_graph(self, pages): gtm = GraphTaskMapper() for page in pages: page.set_sentences_from_text() page.set_tasks_from_sentences() for task in page.tasks: gtm.add_node_and_edge_with_task(task) remover = TaskGraphNodeRemover(gtm.graph) remover.remove_low_score_generalized_tasks() return gtm.graph
def __init__(self, graph=False, query_task='部屋_掃除する'): super().__init__(graph) self.query_task = query_task node_remover = TaskGraphNodeRemover(graph) self.graph = node_remover.graph_without_low_score_generalized_tasks()