Table 3 Summary of parameter settings
Parameter | KIBA | Davis |
|---|---|---|
\(l_d\) | 80 | 36 |
\(l_p\) | 800 | 900 |
\(m^h\) | 0,1,2,3 | 0,1,2,3 |
\(e_d\) | 128 | 64 |
\(L_d\) | 2 | 2 |
\(L_p\) | 2 | 1 |
Hidden size in FFN | 1024,1024,512,1 | |
Epoch | 300 | 300 |
Dropout | 0.1 | 0.1 |
Optimizer | Adam | Adam |
Learning rate | 0.0001 | 0.0001 |