Skip to content

Commit cfe8ecb

Browse files
authored
Fix shape issue in deep rec example (#68)
1 parent 00d9a09 commit cfe8ecb

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/deep_recommender.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -416,9 +416,9 @@ def preprocess_rating(x):
416416
"raw_user_age": features["raw_user_age"],
417417
"user_gender": features["user_gender"],
418418
"user_occupation_label": features["user_occupation_label"],
419-
"user_gender_X_raw_user_age": features[
420-
"user_gender_X_raw_user_age"
421-
],
419+
"user_gender_X_raw_user_age": tf.squeeze(
420+
features["user_gender_X_raw_user_age"], axis=-1
421+
),
422422
# Movie inputs are movie ID, vectorized title and genres
423423
"movie_id": int(x["movie_id"]),
424424
"movie_title_vector": features["movie_title"],

0 commit comments

Comments
 (0)