在线时间:8:00-16:00
迪恩网络APP
随时随地掌握行业动态
扫描二维码
关注迪恩网络微信公众号
在模型选择中我们一般用caret包train函数建立模型,并对模型进行评判 方法1:set.seed(1234) tr_control<-trainControl(method = 'cv',number = 5) # 创建随机森林模型 model_rf<-train(Class~.,data=traindata, trControl=tr_control,method='rf') model_rf 输出 mtry Accuracy Kappa Accuracy was used to select the optimal model using the largest value. 方法2set.seed(1234) model_rf <- train(Class ~., data = traindata, method = 'rf', trControl = trainControl(method = 'cv', number = 5, selectionFunction = 'oneSE')) model_rf mtry Accuracy Kappa Accuracy was used to select the optimal model using the one SE rule.
可以看到二者选定的模型并不一样,而且选定的标准也不一样,方法1标准是最大值法,方法2是精确度。 原因在方法2中用了:selectionFunction = 'oneSE' |
请发表评论