导入图片训练集和测试集后怎样把数据集拆分成x_train、y_train、x_test、y_test
拆分数据集可以使用sklearn库中的train_test_split函数。代码如下:
from sklearn.model_selection import train_test_split
# 假设数据集为data和labels,分别为特征和标签数据
x_train, x_test, y_train, y_test = train_test_split(data, labels, test_size=0.2, random_state=42)
其中,test_size表示测试集所占比例,random_state是随机种子,保证每次运行结果都一样。拆分后,x_train和y_train为训练集特征和标签数据,x_test和y_test为测试集特征和标签数据。
原文地址: http://www.cveoy.top/t/topic/fQWe 著作权归作者所有。请勿转载和采集!