scikit-learn期望2d num数组用于拟合函数的训练数据集。您传入的数据集是一个3d数组,您需要将其重塑为2d形状。
nsamples, nx, ny = train_dataset.shaped2_train_dataset = train_dataset.reshape((nsamples,nx*ny))

scikit-learn期望2d num数组用于拟合函数的训练数据集。您传入的数据集是一个3d数组,您需要将其重塑为2d形状。
nsamples, nx, ny = train_dataset.shaped2_train_dataset = train_dataset.reshape((nsamples,nx*ny))