Scikit通过GuassianNB预测内存错误

Question

我正在使用Scass-kit的GuassianNB学习我的分类。在拟合数据和预测之后，它会引发内存错误。

clf1=GaussianNB() 
clf1.fit(X_train,y_train)
y_pred1=clf1.predict(imgarray)

其中：

1. X_train是一个大小为（1413,2）
的数组
2. y_train是一个大小为（1413，）的数组
3. imgarray大小是（9000000,2）

错误：



其他详细信息：

SCi-Kit学习版：0.15，Windows 7 32位，Python 2.7，pydev，RAM 4 GB

我试图更改版本和其他内容但问题仍在继续。我的imgarray太大了吗？我会感谢你的帮助和建议。

Answer 1

我认为imgarray本身不足以使4GB机器崩溃：

In [3]: a = np.zeros((9000000,2))
In [4]: a.nbytes
Out[4]: 144000000

大约137​​MB。你持有任何其他大型阵列是内存吗？没有看你的代码就很难分辨。你能发布一个完整的可运行的代码片段，以便我们仔细看看吗？

此外，您可以查看此问题以了解如何执行memory profiling。