From 16d6bd7c38d91c2fe8e7b4d5f46902233a5913a8 Mon Sep 17 00:00:00 2001 From: dzhwinter Date: Mon, 8 May 2017 21:43:01 +0800 Subject: [PATCH] "fix label genenerate type. avoid IVector create error when init label" --- python/paddle/v2/dataset/mq2007.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/paddle/v2/dataset/mq2007.py b/python/paddle/v2/dataset/mq2007.py index 5705ba60de..9dd33a22ba 100644 --- a/python/paddle/v2/dataset/mq2007.py +++ b/python/paddle/v2/dataset/mq2007.py @@ -154,9 +154,9 @@ def gen_pair(querylist, partial_order="full"): for i, query_left in enumerate(querylist): for j, query_right in enumerate(querylist): if query_left.relevance_score > query_right.relevance_score: - yield np.ones(1), np.array(query_left.feature_vector), np.array(query_right.feature_vector) + yield 1, np.array(query_left.feature_vector), np.array(query_right.feature_vector) else: - yield np.ones(1), np.array(query_left.feature_vector), np.array(query_right.feature_vector) + yield 1, np.array(query_left.feature_vector), np.array(query_right.feature_vector) elif partial_order == "neighbour": # C(n) @@ -165,9 +165,9 @@ def gen_pair(querylist, partial_order="full"): query_left = querylist[k] query_right = querylist[k+1] if query_left.relevance_score > query_right.relevance_score: - yield np.ones(1), np.array(query_left.feature_vector), np.array(query_right.feature_vector) + yield 1, np.array(query_left.feature_vector), np.array(query_right.feature_vector) else: - yield np.ones(1), np.array(query_left.feature_vector), np.array(query_right.feature_vector) + yield 1, np.array(query_left.feature_vector), np.array(query_right.feature_vector) k += 1 else: raise ValueError("unsupport parameter of partial_order, Only can be neighbour or full")