Python scikit学习MNIST的线性回归内存溢出

我正在尝试使用scikit学习模块对MNIST数据进行基本线性回归。它似乎和MemoryError一起崩溃了。我做错了什么？训练数据集的形状是（60000728）

---

问题在于

sklearn

旧版本的sklearn的实现存在此问题，旧版本在资源管理方面存在问题。尝试升级

sklearn

---

另一个可行的选择是在或中运行此代码。

您发布的代码在Google Colab中运行正常。您（实际上不是）报告的错误无法复制。@谢谢您的签出！可能我的笔记本电脑变旧了。。。

import numpy as np
from tensorflow.keras.datasets import mnist
from sklearn import linear_model

(xTrain, yTrain), (xTest, yTest) = mnist.load_data()

xTrain2D = xTrain.reshape((len(xTrain), -1))
xTest2D = xTest.reshape((len(xTest), -1))

reg = linear_model.LinearRegression()
reg.fit(xTrain2D, yTrain)