使用scikit时SGDClassifier内核死亡

时间:2016-09-27 09:23:47

标签: python scikit-learn

我尝试了这个非常简单的例子

import numpy as np
from sklearn import linear_model
X = np.array([[-1, -1], [-2, -1], [1, 1], [2, 1]])
Y = np.array([1, 1, 2, 2])
clf = linear_model.SGDClassifier()
clf.fit(X, Y)

但内核立即死亡

Kernel died, restarting

安装随机森林没有问题:

from sklearn import ensemble
clf2 = ensemble.RandomForestClassifier()
clf2.fit(X, Y)

我不知道为什么会这样。我正在使用当前版本的anaconda3以及ubuntu 16.04 64位系统上的所有更新。

更新:我刚发现这种情况发生在linear_model

的所有模型中

UPDATE2:MKL就是问题,如下所述:https://github.com/scikit-learn/scikit-learn/issues/5046

conda install nomkl修复了它。

2 个答案:

答案 0 :(得分:0)

您可以更新scikit-learn软件包。 如果你使用像这样的conda使用Anaconda更新:

conda update scikit-learn

如果这不是'解决问题

pip install -U scikit-learn

答案 1 :(得分:0)

MKL是问题所在,如此处所述:https://github.com/scikit-learn/scikit-learn/issues/5046

conda install nomkl对其进行修复。