ROC

結(jié)果

源數(shù)據(jù):鳶尾花數(shù)據(jù)集（僅采用其中的兩種類別的花進(jìn)行訓(xùn)練和檢測）

Summary

features：[’sepal length (cm)’, ’sepal width (cm)’, ’petal length (cm)’, ’petal width (cm)’]

實(shí)例：[5.1, 3.5, 1.4, 0.2]target：’setosa’ ’versicolor’ （0 , 1）

采用回歸方法進(jìn)行擬合得到參數(shù)和bias

model.fit(data_train, data_train_label)

對測試數(shù)據(jù)進(jìn)行預(yù)測得到概率值

res = model.predict(data[:100])

與訓(xùn)練集labels匹配后進(jìn)行排序（從大到小）

pred labels68 0.758208 187 0.753780 176 0.745833 150 0.743156 165 0.741676 175 0.739117 162 0.738255 154 0.737036 152 0.733625 177 0.728139 186 0.727547 174 0.726261 158 0.725150 171 0.724719 136 0.724142 014 0.723990 031 0.721648 041 0.720308 072 0.717723 179 0.712833 197 0.705148 151 0.702838 135 0.702203 098 0.701731 192 0.701106 182 0.700661 153 0.700465 118 0.699350 016 0.696915 064 0.693333 1.. ... ...33 0.658937 096 0.656761 130 0.656279 057 0.655673 14 0.652616 085 0.648620 159 0.648586 119 0.646965 070 0.646262 188 0.644482 18 0.643191 038 0.642704 03 0.640933 055 0.640630 147 0.640444 095 0.639552 113 0.639050 022 0.638485 029 0.635590 090 0.634376 137 0.632224 06 0.631119 046 0.630037 011 0.629718 066 0.627807 142 0.624795 044 0.621703 032 0.614932 024 0.603874 084 0.603249 1

計(jì)算訓(xùn)練集中正確的個(gè)數(shù)與非正確的個(gè)數(shù)

運(yùn)用一下的算式進(jìn)行TPR和FPR的計(jì)算

index = 0for i in sorted_mat.values: if i[1] == 0: TPR.append(TPR[index]) FPR.append(FPR[index] + 1.0/F_num) else: TPR.append(TPR[index] + 1.0/T_num) FPR.append(FPR[index]) index += 1

最后進(jìn)行TPR和FPR的描繪

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持好吧啦網(wǎng)。