Repository navigation

Adding UniSRec model implemented on lightweight class hierarchy with pytorch preprocessing - #306

Merged

feldlime merged 26 commits into

MTSWebServices:mainfrom

TOPAPEC:feat/unisrec-model

Jun 5, 2026

Commits on Apr 22, 2026
- feat: add fast_transformers module with FlatSASRec and UniSRec models
  TOPAPEC
  committed
  d4c9df0
- feat: make UniSRec fully configurable
  TOPAPEC
  committed
  6c875b3
Commits on Apr 24, 2026
- Fast gpu preprocessing and good metrics
  TOPAPEC
  committed
  3cec1e0
- Tests + comparison
  TOPAPEC
  committed
  aa015f8
- add changelog, fixed gpu model load
  TOPAPEC
  committed
  e24fec3
- Formatting
  TOPAPEC
  committed
  7d3850b
Commits on May 14, 2026
- feat: add ONNX export, hash ID mapping, and map_item_ids
  
  TOPAPEC
  authored and
  RecTools Dev
  committed
  f2fdfe5
- Simplify UniSRec: remove 3-phase training, hash IDs, ranking.py
  
  TOPAPEC
  authored and
  RecTools Dev
  committed
  6809160
- Clean up UniSRec: remove dead code, add GPU metrics
  
  TOPAPEC
  authored and
  RecTools Dev
  committed
  45ed8ae
Commits on May 15, 2026
Commits on May 18, 2026
- fix: pass all 7 repo linters for fast_transformers module
  TOPAPEC
  committed
  13e71a8
- docs: add UniSRec tutorial notebook with ML-20M dataset
  TOPAPEC
  committed
  65ef231
Commits on May 19, 2026
- Delete rectools/fast_transformers/unisrec/demo_kion.md
  TOPAPEC
  authored
  204a825
Commits on Jun 4, 2026
- Address PR #306 review: rename nets, fix security/checkpoint/ONNX/predict, add docstrings
  TOPAPEC
  committed
  248bae0
- Remove stale xfail on test_different_sequence_lengths (dynamic_axes works now)
  TOPAPEC
  committed
  16df93f
- Fix black formatting in test_unisrec_model.py
  TOPAPEC
  committed
  56e3a27
- Fix N806 lint ignore: move from global tests/* to local test_metrics.py only
  TOPAPEC
  committed
  39b89d4
- Add type: ignore[import-untyped] for requests in benchmark script
  TOPAPEC
  committed
  3aaef75
- Fix remaining review items: make_ffn docstring, N812 violation, padding test
  TOPAPEC
  committed
  54bb866
- Remove stale type: ignore comments flagged by mypy
  TOPAPEC
  committed
  8ffd648
Commits on Jun 5, 2026
- fix: mypy type errors for requests import and Trainer callbacks
  TOPAPEC
  committed
  63898f7

Navigation Menu

Repository navigation

Adding UniSRec model implemented on lightweight class hierarchy with pytorch preprocessing - #306

Adding UniSRec model implemented on lightweight class hierarchy with pytorch preprocessing #306
feldlime merged 26 commits into
MTSWebServices:mainfrom
TOPAPEC:feat/unisrec-model

Commits on Apr 22, 2026

feat: add fast_transformers module with FlatSASRec and UniSRec models

feat: make UniSRec fully configurable

Commits on Apr 24, 2026

Fast gpu preprocessing and good metrics

Tests + comparison

add changelog, fixed gpu model load

Formatting

Commits on May 14, 2026

feat: add ONNX export, hash ID mapping, and map_item_ids

Simplify UniSRec: remove 3-phase training, hash IDs, ranking.py

Clean up UniSRec: remove dead code, add GPU metrics

Commits on May 15, 2026

fix: address review blockers and majors for UniSRecModel

fix: lint, val+non-softmax loss, device-aware negatives, align_embeddings, n_negatives validation

fix: remove unnecessary dtype preservation in align_embeddings

feat: use GPU for build_sequences in fit() when CUDA available

fix: explicit device param instead of auto-CUDA in fit()

fix: pass all 7 repo linters (mypy, isort, black, flake8, codespell, pylint, bandit)

Commits on May 18, 2026

fix: pass all 7 repo linters for fast_transformers module

docs: add UniSRec tutorial notebook with ML-20M dataset

Commits on May 19, 2026

Delete rectools/fast_transformers/unisrec/demo_kion.md

Commits on Jun 4, 2026

Address PR #306 review: rename nets, fix security/checkpoint/ONNX/predict, add docstrings

Remove stale xfail on test_different_sequence_lengths (dynamic_axes works now)

Fix black formatting in test_unisrec_model.py

Fix N806 lint ignore: move from global tests/* to local test_metrics.py only

Add type: ignore[import-untyped] for requests in benchmark script

Fix remaining review items: make_ffn docstring, N812 violation, padding test

Remove stale type: ignore comments flagged by mypy

Commits on Jun 5, 2026

fix: mypy type errors for requests import and Trainer callbacks