7

我正在尝试使用 sklearn 0.11 的 LogisticRegression 对象来拟合具有大约 80,000 个特征的 200,000 个观察值的模型。目标是将短文本描述分类为 800 个类别中的 1 个。

当我尝试拟合分类器时，pythonw.exe 给了我：

应用程序错误“...处的指令引用了 0x00000000 处的内存”。内存无法写入”。

这些特征非常稀疏，每次观察大约 10 个，并且是二进制的（1 或 0），所以通过我的信封计算，我的 4 GB RAM 应该能够处理内存需求，但这似乎并不就是这样。只有当我使用较少的观察和/或较少的特征时，这些模型才适合。

如果有的话，我想使用更多的观察和特征。我天真的理解是，在幕后运行的 liblinear 库能够支持这一点。关于如何挤进更多观察的任何想法？

我的代码如下所示：

y_vectorizer = LabelVectorizer(y) # my custom vectorizer for labels
y = y_vectorizer.fit_transform(y)

x_vectorizer = CountVectorizer(binary = True, analyzer = features)
x = x_vectorizer.fit_transform(x)

clf = LogisticRegression()
clf.fit(x, y)

我传递给分析器的 features() 函数只返回一个字符串列表，指示在每个观察中检测到的特征。

我正在使用 Python 2.7、sklearn 0.11、具有 4 GB RAM 的 Windows XP。

4

1 回答 1

23

liblinear（的后备实现sklearn.linear_model.LogisticRegression）将托管它自己的数据副本，因为它是一个 C++ 库，其内部内存布局不能直接映射到 scipy 中预先分配的稀疏矩阵，例如scipy.sparse.csr_matrixor scipy.sparse.csc_matrix。

在您的情况下，我建议将您的数据加载为 ascipy.sparse.csr_matrix并将其提供给 a sklearn.linear_model.SGDClassifier（loss='log'如果您想要一个逻辑回归模型并能够调用该predict_proba方法）。如果它已经在使用内存布局SGDClassifier，则不会复制输入数据。scipy.sparse.csr_matrix

期望它在内存中分配 800 * (80000 + 1) * 8 / (1024 ** 2) = 488MB 的密集模型（除了输入数据集的大小）。

编辑：如何优化数据集的内存访问

要在数据集提取后释放内存，您可以：

x_vectorizer = CountVectorizer(binary = True, analyzer = features)
x = x_vectorizer.fit_transform(x)
from sklearn.externals import joblib
joblib.dump(x.tocsr(), 'dataset.joblib')

然后退出这个python进程（强制完成内存释放）并在一个新进程中：

x_csr = joblib.load('dataset.joblib')

在 linux / OSX 下，您可以通过以下方式更有效地进行内存映射：

x_csr = joblib.load('dataset.joblib', mmap_mode='c')
于 2012-06-26T08:53:03.313 回答