malllabiisc/InteractE

Number of parameters

Demirrr opened this issue · 2 comments

Demirrr commented

Hello,

I was wondering where the number of parameters of the InteractE on WN18RR, FB15k-237 and YAGO3-10 is reported somewhere.

Cheers

soumyasanyal commented

Hi Demirrr,

We do a comparison of the score functions of different embedding methods in Table 1 of our paper. We did not report the number of parameters for our best models. If you require them, it should be easy to run the given command to instantiate the model and compute the number of params before this line.

  • 👍1
svjan5 commented

The function to count the number of parameters is given here:
https://gist.github.com/zackenton/12a86b6e0ff274b39608e40f4a412f2b