首页 > 解决方案 > R-Caret 中指定网格上度量的可视化

问题描述

为 SVM 运行 train 函数并指定网格时:

ctrl <- trainControl(method = "repeatedcv", number = 10, repeats = 50, verboseIter = TRUE,
                       savePredictions = TRUE, classProbs = TRUE, summaryFunction = MySummary)
  SVMgrid <- expand.grid(sigma = seq(0.0025,0.03,0.0025), C = seq(0.5,20,2))
  mod_fit <- train(Thrombosis ~ .,  data=df, method="svmRadial", trControl=ctrl,
                   tuneGrid=SVMgrid, preProc=c("scale","YeoJohnson"), metric="Kappa",
                   allowParallel = TRUE)

输出是具有跨指定网格的所需度量值的对象:

> res$model$results
    sigma    C  Accuracy     Kappa
    1   0.0025  0.5 0.9133391 0.1749249
    2   0.0025  2.5 0.9138762 0.2059130
    3   0.0025  4.5 0.9183870 0.2710625
    4   0.0025  6.5 0.9194742 0.2730091
    5   0.0025  8.5 0.9198104 0.2733598
    6   0.0025 10.5 0.9204425 0.2787055
    7   0.0025 12.5 0.9210958 0.2811878
    8   0.0025 14.5 0.9217887 0.2869429
    9   0.0025 16.5 0.9224985 0.2897108
    10  0.0025 18.5 0.9222014 0.2837258
    11  0.0050  0.5 0.9154192 0.2231682

如何将其可视化为正方形网格,或以另一种适当的方式?

标签: rr-caret

解决方案


推荐阅读