Skip to main content
 首页 » 编程设计

python之拆分成 R 中的训练和测试集

2025年05月04日16虾米哥

如何将以下用 python 编写的代码写入 R 中?

X_train, X_test, y_train, y_test = train_test_split(X, y,  
                                                    test_size=0.2, random_state=42)    

按 80/20 的比例分成训练集和测试集。

请您参考如下方法:

可能是更简单的方法

#read in iris dataset  
 data(iris)   
 library(caret) #this package has the createDataPartition function 
     
 set.seed(123) #randomization` 
     
 #creating indices 
 trainIndex <- createDataPartition(iris$Species,p=0.75,list=FALSE) 
     
 #splitting data into training/testing data using the trainIndex object 
 IRIS_TRAIN <- iris[trainIndex,] #training data (75% of data) 
     
 IRIS_TEST <- iris[-trainIndex,] #testing data (25% of data)