如何在scikit-learn中使用字符串内核?

时间:2014-10-15 20:18:25

标签: python string scikit-learn svm

我正在尝试生成一个提供支持向量分类器的字符串内核。我尝试了一个计算内核的函数,就像那样

def stringkernel(K, G):
    for a in range(len(K)):
        for b in range(len(G)):
            R[a][b] = scipy.exp(editdistance(K[a] , G[b]) ** 2)
    return R

当我把它作为参数传递给SVC时我得到

 clf = svm.SVC(kernel = my_kernel)
 clf.fit(data, target)

 ValueError: could not convert string to float: photography

其中我的数据是字符串列表,目标是此字符串所属的对应类。我已经回顾了有关此问题的stackoverflow中的一些问题,但我认为对于这种情况,词汇表示不合适。

2 个答案:

答案 0 :(得分:4)

这是scikit-learn中的一个限制,已经证明很难摆脱。您可以尝试this workaround。用特征向量表示只有一个特征的字符串,这实际上只是字符串表的索引。

>>> data = ["foo", "bar", "baz"]
>>> X = np.arange(len(data)).reshape(-1, 1)
>>> X
array([[0],
       [1],
       [2]])

重新定义字符串内核函数以处理此表示:

>>> def string_kernel(X, Y):
...     R = np.zeros((len(x), len(y)))
...     for x in X:
...         for y in Y:
...             i = int(x[0])
...             j = int(y[0])
...             # simplest kernel ever
...             R[i, j] = data[i][0] == data[j][0]
...     return R
... 
>>> clf = SVC(kernel=string_kernel)
>>> clf.fit(X, ['no', 'yes', 'yes'])
SVC(C=1.0, cache_size=200, class_weight=None, coef0=0.0, degree=3, gamma=0.0,
  kernel=<function string_kernel at 0x7f5988f0bde8>, max_iter=-1,
  probability=False, random_state=None, shrinking=True, tol=0.001,
  verbose=False)

这样做的缺点是,要对新样本进行分类,您必须将它们添加到data,然后为它们构建新的伪特征向量。

>>> data.extend(["bla", "fool"])
>>> clf.predict([[3], [4]])
array(['yes', 'no'], 
      dtype='|S3')

(你可以通过对伪特征进行更多解释来解决这个问题,例如,查看i >= len(X_train)的不同表格。但它仍然很麻烦。)

这是一个丑陋的黑客,但它有效(它对于群集来说稍微不那么难看,因为数据集在fit之后没有变化)。代表scikit-learn开发人员发言,我说一个修复此问题的补丁是受欢迎的。

答案 1 :(得分:1)

我认为shogun库可能是解决方案,也是免费和开源的,我建议查看这个例子:https://github.com/shogun-toolbox/shogun/tree/develop/src/shogun/kernel/string