python-3.x - 尝试训练逻辑回归模型时出现内存故障

Question

我遇到了这个失败:

文件“C:\Users\ophirbh\AppData\Roaming\Python\Python38\site-packages\scipy\special_logsumexp.py”，第 112 行，在 logsumexp 中 tmp = np.exp(a - a_max)MemoryError:无法分配形状为 (950028, 45) 且数据类型为 float64 的数组

对于该代码:

from sklearn import datasets
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import cross_val_score

X, y = datasets.load_svmlight_file('converted_features')
clf = LogisticRegression(random_state = 0, solver = 'lbfgs', multi_class = 
'multinomial',max_iter=2).fit(X,y)
print(cross_val_score(clf, X, y, scoring='recall_macro', cv = 5))

有什么办法可以解决吗？

Answer 1

这种情况你好像内存不足，可以使用google colab用于快速实验，而不会遇到任何内存问题。