From 4f0291be8fc4d795de3a1ad416c02a14c5c6c04c Mon Sep 17 00:00:00 2001 From: Elkana Baris <104759389+ElkanaBaris@users.noreply.github.com> Date: Wed, 19 Apr 2023 11:41:54 +0300 Subject: [PATCH] Update abstract_recommender.py removing unused argument --- recbole/model/abstract_recommender.py | 1 - 1 file changed, 1 deletion(-) diff --git a/recbole/model/abstract_recommender.py b/recbole/model/abstract_recommender.py index 8d7ecbd16..fc35149d1 100644 --- a/recbole/model/abstract_recommender.py +++ b/recbole/model/abstract_recommender.py @@ -335,7 +335,6 @@ def embed_float_fields(self, float_fields): Args: float_fields (torch.FloatTensor): The input dense tensor. shape of [batch_size, num_float_field] - embed (bool): Return the embedding of columns or just the columns itself. Defaults to ``True``. Returns: torch.FloatTensor: The result embedding tensor of float columns.