(self, corpus, model)
| 142 | x = self.nodes[node]['label'] |
| 143 | return x |
| 144 | def GetEmbedding(self, corpus, model): |
| 145 | tmp = [model.wv[i] for i in corpus] |
| 146 | r = np.mean(tmp,axis=0) |
| 147 | return r.tolist() |
| 148 | # def Gettfidf(self,node_attr_vec): |
| 149 | |
| 150 | # cos = cosine_similarity([node_attr_vec],list(self.tfidf.values())) |
no outgoing calls
no test coverage detected