We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 00d9a09 commit cfe8ecbCopy full SHA for cfe8ecb
examples/deep_recommender.py
@@ -416,9 +416,9 @@ def preprocess_rating(x):
416
"raw_user_age": features["raw_user_age"],
417
"user_gender": features["user_gender"],
418
"user_occupation_label": features["user_occupation_label"],
419
- "user_gender_X_raw_user_age": features[
420
- "user_gender_X_raw_user_age"
421
- ],
+ "user_gender_X_raw_user_age": tf.squeeze(
+ features["user_gender_X_raw_user_age"], axis=-1
+ ),
422
# Movie inputs are movie ID, vectorized title and genres
423
"movie_id": int(x["movie_id"]),
424
"movie_title_vector": features["movie_title"],
0 commit comments