From c00a6553c42c183ed364f3791f70d2e30bd4bc5c Mon Sep 17 00:00:00 2001 From: malin10 Date: Wed, 19 Aug 2020 10:56:58 +0800 Subject: [PATCH] update --- models/recall/gru4rec/model.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/models/recall/gru4rec/model.py b/models/recall/gru4rec/model.py index 27e47f1c..d8ebd3bd 100644 --- a/models/recall/gru4rec/model.py +++ b/models/recall/gru4rec/model.py @@ -16,7 +16,7 @@ import paddle.fluid as fluid from paddlerec.core.utils import envs from paddlerec.core.model import ModelBase -from paddlerec.core.metrics import Precision +from paddlerec.core.metrics import RecallK class Model(ModelBase): @@ -82,15 +82,13 @@ class Model(ModelBase): high=self.init_high_bound), learning_rate=self.fc_lr_x)) cost = fluid.layers.cross_entropy(input=fc, label=dst_wordseq) - # acc = fluid.layers.accuracy( - # input=fc, label=dst_wordseq, k=self.recall_k) - acc = Precision(input=fc, label=dst_wordseq, k=self.recall_k) + acc = RecallK(input=fc, label=dst_wordseq, k=self.recall_k) if is_infer: - self._infer_results['P@20'] = acc + self._infer_results['Recall@20'] = acc return avg_cost = fluid.layers.mean(x=cost) self._cost = avg_cost self._metrics["cost"] = avg_cost - self._metrics["P@20"] = acc + self._metrics["Recall@20"] = acc -- GitLab