Skip to content

Commit 9b5adfa

Browse files
authored
Merge pull request #2905 from dzhwinter/fix_mq2007
"fix dense vector shape in mq2007"
2 parents 23b8346 + 56fbed9 commit 9b5adfa

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

python/paddle/v2/dataset/mq2007.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -212,19 +212,19 @@ def gen_pair(querylist, partial_order="full"):
212212
for j in range(i + 1, len(querylist)):
213213
query_right = querylist[j]
214214
if query_left.relevance_score > query_right.relevance_score:
215-
labels.append(1)
215+
labels.append([1])
216216
docpairs.append([
217217
np.array(query_left.feature_vector),
218218
np.array(query_right.feature_vector)
219219
])
220220
elif query_left.relevance_score < query_right.relevance_score:
221-
labels.append(1)
221+
labels.append([1])
222222
docpairs.append([
223223
np.array(query_right.feature_vector),
224224
np.array(query_left.feature_vector)
225225
])
226226
for label, pair in zip(labels, docpairs):
227-
yield label, pair[0], pair[1]
227+
yield np.array(label), pair[0], pair[1]
228228

229229

230230
def gen_list(querylist):

0 commit comments

Comments
 (0)