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.