From 87c912d483075951b326292664f2627fbf8310dd Mon Sep 17 00:00:00 2001 From: dongdaxiang Date: Sat, 18 Apr 2020 19:33:04 +0800 Subject: [PATCH] fix test_client.py with numpy output --- python/examples/criteo_ctr_with_cube/test_client.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/examples/criteo_ctr_with_cube/test_client.py b/python/examples/criteo_ctr_with_cube/test_client.py index de205ebc..ca752b76 100755 --- a/python/examples/criteo_ctr_with_cube/test_client.py +++ b/python/examples/criteo_ctr_with_cube/test_client.py @@ -40,7 +40,7 @@ for ei in range(10000): for i in range(1, 27): feed_dict["embedding_{}.tmp_0".format(i - 1)] = data[0][i] fetch_map = client.predict(feed=feed_dict, fetch=["prob"]) - prob_list.append(fetch_map['prob'][1]) + prob_list.append(fetch_map['prob'][0][1]) label_list.append(data[0][-1][0]) print(auc(label_list, prob_list)) -- GitLab