函数返回值(R)

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])时，它报告为list。似乎模型文件被转换成了一个列表。如何保留原始数据类型而不将其转换为列表，因为我需要在返回时访问模型文件。

Answer 1

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

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

因此，您的函数似乎运行良好，只需确保使用output<-trainer(...); output[[1]]提取模型即可。