Skip to content

API redefinition and performance improvement

Compare
Choose a tag to compare
@armand33 armand33 released this 05 May 09:30
· 325 commits to master since this release
e4874be
  • Redefinition of the models' API (simplified interfaces, renamed LP methods and added get_embeddings method)
  • Implementation of the new API for all models
  • TorusE implementation fixed
  • TransD reimplementation to avoid matmul usage (costly in back-propagation)
  • Bilinear models performances improved by avoiding matmul usage
  • Addition of feature to negative samplers to generate several negative samples from each fact. Those can be fed directly to
    the models.
  • Addition some wrappers for training to utils module.
  • Progress bars now make the most of tqdm's possibilities
  • Code reformatting
  • Docstrings update