如何畫ROC曲線

假設我們現在要畫一個已經訓練好的二分類模型A的ROC曲線。 如下是20個測試樣本,第一列代表樣本id,第二列代表他們的真實類別(p表示正樣本,n表示負樣本),最後一列代表訓練好的模型A認爲每個樣本是正樣本的概率。 下面就是此訓練好的模型A的ROC曲線的畫法: 第一步:按照屬於‘正樣本’的概率將所有樣本排序(如上圖所示) 第二步:把分類閾值設爲最大,即把所有樣例均預測爲反例,所以此時得到P和R均爲0
相關文章
相關標籤/搜索