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:
'\n Generate predictions by defining the architecture\n :return:\n '
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)
user_item_vector_concat = tf.keras.layers.concatenate([input_item_vector_reshaped, input_user_vector_reshaped], axis=1)
dense1 = tf.keras.layers.Dense(self.hidden1_dim)(user_item_vector_concat)
dropout_1 = tf.keras.layers.Dropout(0.1)(dense1)
dense2 = tf.keras.layers.Dense(self.hidden2_dim)(dropout_1)
predicted_rating = tf.keras.layers.Dense(1, activation='linear')(dense2)
return predicted_rating
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"