Add new LTR model

This commit is contained in:
Daoud Clarke 2022-08-09 22:47:59 +01:00
parent 57476ed2c8
commit c1b9e70743
2 changed files with 1 additions and 0 deletions

Binary file not shown.

View File

@ -27,6 +27,7 @@ class ThresholdPredictor(BaseEstimator, RegressorMixin):
def get_features_as_series(item: Series):
terms = item['query'].lower().split()
features = get_features(terms, item['title'], item['url'], item['extract'], item['score'], True)
# features_filtered = {k: v for k, v in features.items() if 'match_score' not in k}
return Series(features)