Question

trainer <- function(training, testing){    
  trControl <- trainControl(method="cv", number=5)
  modelFit <- train(training$classe ~ ., method="rf", preProcess="pca", 
                    trControl=trControl, data=training)
  confMatrix <- confusionMatrix(testing$classe, predict(modelFit,testing))

  output <- list(modelFit, confMatrix)
  return(output)
}

返回的值应该是模型modelFit，它不是列表，但是当我检查class(output[1])时，它会报告为列表。似乎某种程度上模型文件被转换为列表。如何保留原始数据类型而不将其转换为列表，因为我需要在返回时访问模型文件。

Answer 1

如上所述，当output是一个列表并且您使用[ ]来提取元素时，您将始终返回一个列表（包含您指定的索引的元素或元素）。但是，当您使用[[ ]]时，您将在特定位置提取元素，并且该元素可能具有不同的类。

您可以sapply(output, class)查看给定列表中所有项目的类别。

因此，您的功能似乎运行正常，只需确保使用output<-trainer(...); output[[1]]来提取模型。

R中的函数返回值

1 个答案: