File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff 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
230230def gen_list (querylist ):
You can’t perform that action at this time.
0 commit comments