Skip to content

0.13.0

Choose a tag to compare

View all tags
@blondered blondered released this 10 Apr 14:41
· 15 commits to main since this release
v0.13.0
6ffd25b
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode.

✨ Highlights ✨

Transformer models get more customization options, including negative sampling, similarity functions and backbone model. Sampled softmax loss is now supported by all models and cosine similarity function is now available out of the box.

Added

  • TransformerNegativeSamplerBase and CatalogUniformSampler classes, negative_sampler_type and negative_sampler_kwargs parameters to transformer-based models (#275)
  • SimilarityModuleBase, DistanceSimilarityModule, similarity module to TransformerTorchBackbone parameters to transformer-based models similarity_module_type, similarity_module_kwargs (#272)
  • TransformerBackboneBase, backbone_type and backbone_kwargs parameters to transformer-based models (#277)
  • sampled_softmax loss option for transformer models (#274)
  • out_dim property to IdEmbeddingsItemNet, CatFeaturesItemNet and SumOfEmbeddingsConstructor (#276)

All contributors

@spirinamayya @In48semenov @blondered

Contributors

In48semenov, spirinamayya, and blondered
Assets 2
Loading