From c6a349b682df90907fab004c9e2c089c7134deb5 Mon Sep 17 00:00:00 2001 From: lthoang Date: Fri, 24 Nov 2023 20:06:37 +0800 Subject: [PATCH] refactor code --- cornac/eval_methods/next_basket_evaluation.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/cornac/eval_methods/next_basket_evaluation.py b/cornac/eval_methods/next_basket_evaluation.py index e6ed9b81e..fc58af141 100644 --- a/cornac/eval_methods/next_basket_evaluation.py +++ b/cornac/eval_methods/next_basket_evaluation.py @@ -41,18 +41,18 @@ def ranking_eval( Parameters ---------- - model: :obj:`cornac.models.Recommender`, required - Recommender model to be evaluated. + model: :obj:`cornac.models.BasketRecommender`, required + BasketRecommender model to be evaluated. metrics: :obj:`iterable`, required List of rating metrics :obj:`cornac.metrics.RankingMetric`. - train_set: :obj:`cornac.data.Dataset`, required - Dataset to be used for model training. This will be used to exclude + train_set: :obj:`cornac.data.BasketDataset`, required + BasketDataset to be used for model training. This will be used to exclude observations already appeared during training. - test_set: :obj:`cornac.data.Dataset`, required - Dataset to be used for evaluation. + test_set: :obj:`cornac.data.BasketDataset`, required + BasketDataset to be used for evaluation. repetition_eval: boolean, optional, Evaluation on repetition items, appeared in history baskets.