Wie erstellt man eine Kreuzvalidierung falten Variable über createFolds Funktion in Caret?

Question

Wie erstelle ich eine Variable, falte, in meinem Analysedatensatz, dem die Werte von 1: k aus der createFolds Kreuzvalidierungsfunktion in caret zugewiesen sind?

Zum Beispiel des folgenden College Admissions-Datensatz mit:

# Load data. 
mydata <- read.csv("http://www.ats.ucla.edu/stat/data/binary.csv") 
folds <- createFolds(mydata$admit, k=5) 

# Create variable "mydata$fold" and assign values of 1:5 to mydata$fold. 
? 

Answer 1

Sie können dies tun, mit createFolds, hat es ein Argument list, die standardmäßig TRUE in jeder Falte eine Liste der indicies geben. Wenn Sie es in FALSE ändern, gibt die Funktion einen Vektor der Falten zurück, denen jede Zeile zugewiesen ist.

folds <- createFolds(mydata$admit, k=5,list = FALSE) 

Weitere Informationen finden Sie unter ?createFolds.