(self, embeds1, embeds2)
| 499 | return self.act(embeds @ self.W) + embeds |
| 500 | |
| 501 | def pairPred(self, embeds1, embeds2): |
| 502 | return (self.featureExtract(embeds1) * self.featureExtract(embeds2)).sum(dim=-1) |
| 503 | |
| 504 | def crossPred(self, embeds1, embeds2): |
| 505 | return self.featureExtract(embeds1) @ self.featureExtract(embeds2).T |
no test coverage detected