(self, corpus, model)
| 110 | x = -1 |
| 111 | return x |
| 112 | def GetEmbedding(self, corpus, model): |
| 113 | tmp = [model.wv[i] for i in corpus] |
| 114 | r = np.mean(tmp,axis=0) |
| 115 | return r.tolist() |
| 116 | # def Gettfidf(self,node_attr_vec): |
| 117 | |
| 118 | # cos = cosine_similarity([node_attr_vec],list(self.tfidf.values())) |
no outgoing calls
no test coverage detected