首页 > 解决方案 > 如何使用 sklearn 弹性网制作 ROC?

问题描述

我有:

r = 6 个特征的 numpy 数组,即形状 ~10000 x 6

答案 = 我想学习预测的二进制 numpy 数组(1 或 0),形状 ~1000 x 1

我已经搞定了:

[datatrain, datatest, answertrain, answertest] = cross_validation.train_test_split(r,answer)
clf = linear_model.ElasticNet()
clf.fit(X=datatrain,y=answertrain) 
prediction = clf.predict(datatest)

如何获得 ROC 的阈值?我没有看到决策函数或预测概率或任何东西的属性。

标签: pythonmachine-learningscikit-learn

解决方案


推荐阅读