Skip to content

Commit aaa047e

Browse files
default validation k cleaned
1 parent dc31120 commit aaa047e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

elliot/recommender/base_recommender_model.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def __init__(self, data, config, params, *args, **kwargs):
3535
_cutoff_k = getattr(data.config.evaluation, "cutoffs", [data.config.top_k])
3636
_cutoff_k = _cutoff_k if isinstance(_cutoff_k, list) else [_cutoff_k]
3737
_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]
38+
_default_validation_k = _cutoff_k[0]
3939
self._validation_metric = getattr(self._params.meta, "validation_metric",
4040
_first_metric + "@" + str(_default_validation_k)).split("@")
4141
if self._validation_metric[0].lower() not in [m.lower()

0 commit comments

Comments
 (0)