重回帰分析の自作関数が上手く作動しません

Question

mlm_func <- function(A){
  y <- colnames(A)[1]
  fit.mlm <- train(A$y ~ ., data = A,　method = "lm",　trControl = trainControl(method = "cv", number = 5))
  mlm.pred <- predict(fit.mlm, testdata)
  RMSE <- sqrt(sum((testdata$y - mlm.pred)^2) / 100 )
  }
dat <- A[,c(1,2,6)]
mlm_func(dat)

重回帰分析を毎回変数を選択して同じ式に代入するのが億劫だったので自作関数を作ってみようとしたのですが、
上手く作動せず以下のエラーが出ます。

 model.frame.default(form = A$y ~ ., data = A, na.action = na.fail) でエラー: 
   型 (NULL) は変数 'dat$y' に対しては不正です 
 追加情報:  警告メッセージ: 
 model.frame.default(form = A$y ~ ., data =　A, na.action = na.fail) でエラー: 
   型 (NULL) は変数 'A$y' に対しては不正です 

どこが間違いなのでしょうか？
また、みなさんがこのような関数を作るときはどうされますか？

Answer 1

おそらく存在しないはずのdat$yとかにアクセスしようとしているのがまずいのでしょう。

まずは、train()の前にbrowser()を追加した上で当該関数の実行を試みると、エラー直前でデバッグモードに入るはずなので、そこで各パラメータが期待した通りのものかチェックしてみるといいでしょう。

Comments

[taka8]

browser()しながらいじってたらできました！
ありがとうございます。