(self)
| 108 | self.save() |
| 109 | |
| 110 | def get_pmi_edge(self): |
| 111 | pmi_edge_lst, self.pmi_time = get_pmi_edge(self.content, window_size=20, threshold=0.0) |
| 112 | print("pmi time:", self.pmi_time) |
| 113 | |
| 114 | for edge_item in pmi_edge_lst: |
| 115 | word_indx1 = self.node_num + self.word2id[edge_item[0]] |
| 116 | word_indx2 = self.node_num + self.word2id[edge_item[1]] |
| 117 | if word_indx1 == word_indx2: |
| 118 | continue |
| 119 | self.g.add_edge(word_indx1, word_indx2, weight=edge_item[2]) |
| 120 | |
| 121 | print_graph_detail(self.g) |
| 122 | |
| 123 | def get_tfidf_edge(self): |
| 124 | # 获得tfidf权重矩阵（sparse）和单词列表 |
no test coverage detected