Skip to content

0.11.0

Compare
Choose a tag to compare
@blondered blondered released this 17 Feb 15:58
· 16 commits to main since this release
e8728b3

✨ Highlights ✨

Transformer models are here!

BERT4Rec and SASRec are fully compatible with RecTools fit / recommend paradigm and require NO special data processing. We have proven top performance on public benchmarks. For details on models see Transformers Theory & Practice Tutorial.

Our transformer models are configurable, customizable, callback-friendly, checkpoints-included, logs-out-of-the-box, custom-validation-ready, multi-gpu-compatible! See Transformers Advanced Training Guide and Transformers Customization Guide

All updates

Added

  • SASRecModel and BERT4RecModel - models based on transformer architecture (#220)
  • Transfomers extended theory & practice tutorial, advanced training guide and customization guide (#220)
  • use_gpu for PureSVD (#229)
  • from_params method for models and model_from_params function (#252)
  • TorchRanker ranker which calculates scores using torch. Supports GPU. #251
  • Ranker ranker protocol which unify rankers call. #251

Changed

  • ImplicitRanker rank method compatible with Ranker protocol. use_gpu and num_threads params moved from rank method to __init__. #251

New contributors

@nsundalov made their first contribution in #251

All contributors

@feldlime @blondered @spirinamayya @chezou @In48semenov