We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent dc31120 commit aaa047eCopy full SHA for aaa047e
elliot/recommender/base_recommender_model.py
@@ -35,7 +35,7 @@ def __init__(self, data, config, params, *args, **kwargs):
35
_cutoff_k = getattr(data.config.evaluation, "cutoffs", [data.config.top_k])
36
_cutoff_k = _cutoff_k if isinstance(_cutoff_k, list) else [_cutoff_k]
37
_first_metric = data.config.evaluation.simple_metrics[0] if data.config.evaluation.simple_metrics else ""
38
- _default_validation_k = getattr(data.config.evaluation, "cutoffs", [data.config.top_k])[0]
+ _default_validation_k = _cutoff_k[0]
39
self._validation_metric = getattr(self._params.meta, "validation_metric",
40
_first_metric + "@" + str(_default_validation_k)).split("@")
41
if self._validation_metric[0].lower() not in [m.lower()
0 commit comments