提交 e5cb9b77 编写于 作者: D dongzhihong

"fix gen list"

上级 afe7bf61
...@@ -242,9 +242,9 @@ def gen_list(querylist): ...@@ -242,9 +242,9 @@ def gen_list(querylist):
if not isinstance(querylist, QueryList): if not isinstance(querylist, QueryList):
querylist = QueryList(querylist) querylist = QueryList(querylist)
querylist._correct_ranking_() querylist._correct_ranking_()
relevance_score_list = [query.relevance_score for query in querylist] relevance_score_list = [[query.relevance_score] for query in querylist]
feature_vector_list = [query.feature_vector for query in querylist] feature_vector_list = [query.feature_vector for query in querylist]
yield np.array(relevance_score_list).T, np.array(feature_vector_list) yield np.array(relevance_score_list), np.array(feature_vector_list)
def query_filter(querylists): def query_filter(querylists):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册