dvarelas / tensorflow2-recommender

MIT License
7 stars 2 forks source link

tf rec issue #13

Open dvarelas opened 1 year ago

dvarelas commented 1 year ago

File "/data/code/tensorflow2-recommender/tf2recommender/models/ncf.py", line 68, in predict user_item_vector_concat = tf.keras.layers.concatenate( [input_item_vector_reshaped, input_user_vector_reshaped], axis=1) InvalidArgumentError: Exception encountered when calling layer "tf.concat_7"

fixobot-404[bot] commented 1 year ago

Here's the suspect file: https://github.com/dvarelas/tensorflow2-recommender/blob/master/tf2recommender/models/ncf.py


and the suspect function:

    def predict(self):
        """
        Generate predictions by defining the architecture
        :return:
        """
        input_item_vector = self.item_embeddings(self.input_item)
        input_user_vector = self.user_embeddings(self.input_user)
        input_item_vector_reshaped = tf.keras.layers.Reshape(
            (self.item_dim, 1))(input_item_vector)
        input_user_vector_reshaped = tf.keras.layers.Reshape(
            (self.user_dim, 1))(input_user_vector)