sklearn交叉验证 - 获得原始预测

时间:2017-06-14 09:29:19

标签: python scikit-learn cross-validation lda

我正在使用scikit学习cross_val_predict函数来对LDA回归进行交叉验证。

虽然我正在进行二元预测,但我想使用此函数来获取" raw"在将它们转换为二进制之前作为小数的预测。

from sklearn.model_selection import ShuffleSplit
from sklearn.model_selection import cross_val_score, cross_val_predict
from sklearn import metrics
from sklearn.discriminant_analysis import LinearDiscriminantAnalysis
clf = LinearDiscriminantAnalysis()
cv = ShuffleSplit(n_splits=5, random_state=1)
scores = cross_val_score(clf, final_list, lab_list, cv=cv, scoring='roc_auc')
pred = cross_val_predict(clf, final_list, lab_list, cv=5) 

目前,pred是二进制列表,而我想要一个十进制输出以进行进一步的统计分析。这可能与使用的功能有关吗?

谢谢!

0 个答案:

没有答案