Open dvarelas opened 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)
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"