diff --git a/05.recommender_system/train.py b/05.recommender_system/train.py index f20a848968ff8b4b9bba87a970933102d4754d9a..7b1f971b0a54f380345fc613d01ae5a24f9cc9eb 100644 --- a/05.recommender_system/train.py +++ b/05.recommender_system/train.py @@ -289,8 +289,8 @@ def infer(use_cuda, params_dirname): category_id = fluid.create_lod_tensor([[10L, 8L, 9L]], [[3]], place) assert feed_target_names[6] == "movie_title" - movie_title = fluid.create_lod_tensor([[1069L, 4140L, 2923L, 710L, 988L]], - [[5]], place) + movie_title = fluid.create_lod_tensor( + [[1069L, 4140L, 2923L, 710L, 988L]], [[5]], place) # Construct feed as a dictionary of {feed_target_name: feed_target_data} # and results will contain a list of data corresponding to fetch_targets.