-
Notifications
You must be signed in to change notification settings - Fork 42
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Fix get_train_with_targets_for_reranker method #244
base: experimental/two_stage
Are you sure you want to change the base?
Fix get_train_with_targets_for_reranker method #244
Conversation
Co-authored-by: Daria <93913290+blondered@users.noreply.github.com>
Co-authored-by: Andrey Semenov <43339130+In48semenov@users.noreply.github.com>
Co-authored-by: Andrey Semenov <43339130+In48semenov@users.noreply.github.com>
Co-authored-by: Andrey Semenov <43339130+In48semenov@users.noreply.github.com>
Co-authored-by: Andrey Semenov <43339130+In48semenov@users.noreply.github.com>
Co-authored-by: Andrey Semenov <43339130+In48semenov@users.noreply.github.com>
Co-authored-by: Andrey Semenov <43339130+In48semenov@users.noreply.github.com>
Co-authored-by: Andrey Semenov <43339130+In48semenov@users.noreply.github.com>
Co-authored-by: Andrey Semenov <43339130+In48semenov@users.noreply.github.com>
…gakova/RecTools into experimental/two_stage
|
||
def test_prepare_training_pool( | ||
self, | ||
reranker_catboost_ranker: CatBoostReranker, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please move ranker and classifier to pytest.parametrize
decorator for all of the tests where you need to test both approaches.
Here is an example on how to organize code when you have different outputs for different parameters:
@pytest.mark.parametrize( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Fixed
Co-authored-by: Andrey Semenov <43339130+In48semenov@users.noreply.github.com>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please fix these mistakes everywhere
We make changes to the
get_train_with_targets_for_reranker
method to separate the retrieval of sampled candidates and unsampled candidates from first-stage candidate generators for the reranker.