refine fluid recomm test

7 years ago · 42bf89ce12
parent 6b47598103
commit 42bf89ce12
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/v2/fluid/tests/book/test_recommender_system.py
+++ b/python/paddle/v2/fluid/tests/book/test_recommender_system.py
@ -125,10 +125,11 @@ def model():

    # need cos sim
    inference = layers.cos_sim(X=usr_combined_features, Y=mov_combined_features)
+    scale_infer = layers.scale(x=inference, scale=5.0)

    label = layers.data(name='score', shape=[1], dtype='float32')

-    square_cost = layers.square_error_cost(input=inference, label=label)
+    square_cost = layers.square_error_cost(input=scale_infer, label=label)

    avg_cost = layers.mean(x=square_cost)