首页 文章

使用具有不同调整参数的R Caret模型

提问于
浏览
1

我训练了NNET Caret模型,其中有三种尺寸的调音功能 . 最终的模型配有一个尺寸 . 它由最小的RMSE自动选择 .

Neural Network 

9700 samples
  23 predictor

Pre-processing: centered, scaled 
Resampling: Cross-Validated (8 fold, repeated 8 times) 

Summary of sample sizes: 8488, 8488, 8487, 8485, 8488, 8488, ... 

Resampling results across tuning parameters:

  size  RMSE    Rsquared  RMSE SD   Rsquared SD
  12    0.0328  0.951     0.002033  0.006127   
  24    0.0221  0.978     0.001358  0.002764   
  72    0.0134  0.992     0.000647  0.000815   

Tuning parameter 'decay' was held constant at a value of 5e-04
RMSE was used to select the optimal model using  the smallest value.
The final values used for the model were size = 72 and decay = 5e-04.

但是我想探索用其他尺寸训练的模型 . 我可以将预测函数与其他模型的参数一起使用吗?

1 回答

  • 2

    如果使用 trainControlsavePredictions 参数,则可以从调整网格中的不同模型中获取保持预测 . 除此之外,您还需要适合单独的模型 .

    马克斯

相关问题