dvarelas / tensorflow2-recommender

MIT License
7 stars 2 forks source link

tf issue no file #14

Open dvarelas opened 1 year ago

dvarelas commented 1 year ago

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: 

'\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