以下是加载digits数据集并拆分成训练集和测试集的代码:

from sklearn.datasets import load_digits
from sklearn.model_selection import train_test_split

# 加载digits数据集
digits = load_digits()

# 拆分数据集
X_train, X_test, y_train, y_test = train_test_split(digits.data, digits.target, test_size=0.2, random_state=666)

# 打印拆分后的数据集形状
print("训练集样本数量:", X_train.shape[0])
print("测试集样本数量:", X_test.shape[0])

运行以上代码后,将得到训练集X_trainy_train,以及测试集X_testy_test。其中,训练集占80%,测试集占20%。并且设置了随机数种子为666。

加载sklearn中datasets模块的digits手写数字数据集使用sklearn中model_selection模块的train_test_split函数将digits手写数字数据集拆分成训练数据集及测试数据集要求测试数据集占20随机数种子设定为666。

原文地址: https://www.cveoy.top/t/topic/h9lK 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录