利用ModelMetrics包计算模型的常用参数

既可以直接用预测值和实际值来计算模型的各个参数,还可以直接把模型放进去直接计算,可以用来计算的模型包括以下: 'glm', 'randomForest', 'glmerMod', 'gbm', 'rpart'。

library(ModelMetrics)

data(testDF)
str(testDF)
测试数据集

可以看到,Y为结局比那里,为二分类,而x1及x2为数值型变量

glmModel <- glm(y ~ ., data = testDF, family="binomial")#构建logistic回归
Preds <- predict(glmModel, type = 'response')#利用predict函数构建预测概率向量
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)
混合矩阵
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • R语言与数据挖掘:公式;数据;方法 R语言特征 对大小写敏感 通常,数字,字母,. 和 _都是允许的(在一些国家还...
    __一蓑烟雨__阅读 1,696评论 0 5
  • 主要涉及树模型 Supported Metrics MSEdevianceRMSEMAER2RMSLE class...
    Liam_ml阅读 1,695评论 0 1
  • 建立模型的目的是提供一个简单的、低纬度的数据集摘要。我们使用模型的目的是将数据划分为模式和残差,借助模型探索数据集...
    王诗翔阅读 4,847评论 0 2
  • 原文 机器学习(Machine Learning, ML)是一门多领域交叉学科,涉及概率论、统计学、逼近论、凸分析...
    readilen阅读 4,017评论 0 41
  • 久违的晴天,家长会。 家长大会开好到教室时,离放学已经没多少时间了。班主任说已经安排了三个家长分享经验。 放学铃声...
    飘雪儿5阅读 7,590评论 16 22