def eval_clf(gold, clf, mat, start = 0):
pp = clf.predict_proba(mat[start:,:])
pp1 = pp[:,1]
ap = average_precision_score(gold[start:], pp1)
return ap
微信
公众号
扫码关注公众号