利用ModelMetrics包計算模型的常用參數(shù)

既可以直接用預測值和實際值來計算模型的各個參數(shù),還可以直接把模型放進去直接計算,可以用來計算的模型包括以下: 'glm', 'randomForest', 'glmerMod', 'gbm', 'rpart'。

library(ModelMetrics)

data(testDF)
str(testDF)
測試數(shù)據(jù)集

可以看到,Y為結(jié)局比那里,為二分類,而x1及x2為數(shù)值型變量

glmModel <- glm(y ~ ., data = testDF, family="binomial")#構(gòu)建logistic回歸
Preds <- predict(glmModel, type = 'response')#利用predict函數(shù)構(gòu)建預測概率向量
head(Preds)
預測概率
auc(testDF$y, Preds)#利用實際值及預測概率計算AUC值
#0.9872666
auc(glmModel)#直接利用回歸方程計算AUC值
#0.9872666
brier(testDF$y, Preds)#利用實際值及預測概率計算brier值
#0.04788846
brier(glmModel)#直接利用回歸方程計算AUC值
#0.04788846

kappa(testDF$y, Preds, cutoff = 0.5)#計算kappa值
#0.8506826
confusionMatrix(testDF$y, Preds, cutoff = 0.5)
混合矩陣
?著作權歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

  • R語言與數(shù)據(jù)挖掘:公式;數(shù)據(jù);方法 R語言特征 對大小寫敏感 通常,數(shù)字,字母,. 和 _都是允許的(在一些國家還...
    __一蓑煙雨__閱讀 1,816評論 0 5
  • 主要涉及樹模型 Supported Metrics MSEdevianceRMSEMAER2RMSLE class...
    Liam_ml閱讀 1,736評論 0 1
  • 建立模型的目的是提供一個簡單的、低緯度的數(shù)據(jù)集摘要。我們使用模型的目的是將數(shù)據(jù)劃分為模式和殘差,借助模型探索數(shù)據(jù)集...
    王詩翔閱讀 5,019評論 0 2
  • 原文 機器學習(Machine Learning, ML)是一門多領域交叉學科,涉及概率論、統(tǒng)計學、逼近論、凸分析...
    readilen閱讀 4,066評論 0 41
  • 久違的晴天,家長會。 家長大會開好到教室時,離放學已經(jīng)沒多少時間了。班主任說已經(jīng)安排了三個家長分享經(jīng)驗。 放學鈴聲...
    飄雪兒5閱讀 7,788評論 16 22

友情鏈接更多精彩內(nèi)容