Skip to content

Commit 41a5a63

Browse files
authored
Experimental/two stage (#296)
Two-stage model added
1 parent 3391786 commit 41a5a63

19 files changed

Lines changed: 5129 additions & 894 deletions

.github/workflows/test.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,10 +29,10 @@ jobs:
2929
with:
3030
path: .venv
3131
key: venv-lint-${{ runner.os }}-${{ hashFiles('**/poetry.lock') }}
32-
32+
3333
- name: Install dependencies
3434
if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true'
35-
run: make install
35+
run: make install
3636

3737
- name: Static analysis
3838
run: make lint
@@ -62,10 +62,10 @@ jobs:
6262
with:
6363
path: .venv
6464
key: venv-test-${{ runner.os }}-${{ matrix.python-version }}-${{ hashFiles('**/poetry.lock') }}
65-
65+
6666
- name: Install dependencies
6767
if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true'
68-
run: make install
68+
run: make install
6969

7070
- name: Run tests
7171
run: make test

.gitignore

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,3 +93,6 @@ benchmark_results/
9393
*.zip
9494
*.csv
9595
*.dat
96+
97+
# CatBoost
98+
catboost_info/

CHANGELOG.md

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,13 @@ All notable changes to this project will be documented in this file.
55
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
66
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
77

8+
9+
## Unreleased
10+
11+
### Added
12+
- Two-stage candidate ranking system with `CandidateRankingModel` and supporting classes (`CandidateGenerator`, `CandidateFeatureCollector`, `Reranker`, `CatBoostReranker`, `PerUserNegativeSampler`) ([#296](https://github.com/MobileTeleSystems/RecTools/pull/296))
13+
14+
815
## [0.17.0] - 03.09.2025
916

1017
### Added

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,7 @@ The default version doesn't contain all the dependencies, because some of them a
9898
- `torch`: adds models based on neural nets,
9999
- `visuals`: adds visualization tools,
100100
- `nmslib`: adds fast ANN recommenders.
101+
- `catboost`: adds CatBoost as a reranker for `CandidateRankingModel`
101102

102103
Install extension:
103104
```

0 commit comments

Comments
 (0)