Stonesjtu / Pytorch-NCE

The Noise Contrastive Estimation for softmax output written in Pytorch
MIT License
317 stars 45 forks source link

Generic nce #4

Closed Stonesjtu closed 6 years ago

Stonesjtu commented 6 years ago

A brand new NCE implementation, with more flexibility to write NCE for not only linear classifier, but any NN structures.