为多标记SVM指定权重以平衡类

Chr*_*rry 2 python machine-learning svm scikit-learn

这是怎么做到的?我正在使用Sklearn来训练SVM.我的班级不平衡.请注意,我的问题是多类,多标签,所以我使用的是OneVsRestClassifier:

mlb = MultiLabelBinarizer()
y = mlb.fit_transform(y_train)

clf = OneVsRestClassifier(svm.SVC(kernel='rbf'))
clf = clf.fit(x, y) 
pred = clf.predict(x_test)
Run Code Online (Sandbox Code Playgroud)

我可以在某处添加'sample_weight'参数来解释不平衡的类吗?


当我向svm添加class_weight dict时,我收到错误:

ValueError: Class label 2 not present
Run Code Online (Sandbox Code Playgroud)

这是因为我使用mlb将我的标签转换为二进制.但是,如果我不转换标签,我会得到:

ValueError: You appear to be using a legacy multi-label data representation. Sequence of sequences are no longer supported; use a binary array or sparse matrix instead. 
Run Code Online (Sandbox Code Playgroud)

class_weight是一个dict,将类标签映射到权重:{1:1,2:1,3:3 ......}

以下是x和y的详细信息:

print(X[0])  
[ 0.76625633  0.63062721  0.01954162 ...,  1.1767817   0.249034    0.23544988]
print(type(X))
<type 'numpy.ndarray'>

print(y[0])
print(type(y))
[1, 2, 3, 4, 5, 6, 7]
<type 'numpy.ndarray'>
Run Code Online (Sandbox Code Playgroud)

注意mlb = MultiLabelBinarizer(); y = mlb.fit_transform(y_train)将y转换为二进制数组.


建议的答案产生错误:

ValueError: You appear to be using a legacy multi-label data representation. Sequence of sequences are no longer supported; use a binary array or sparse matrix instead.
Run Code Online (Sandbox Code Playgroud)

因此,问题减少到将标签(np.array)转换为稀疏矩阵.

from scipy import sparse
y_sp = sparse.csr_matrix(y) 
Run Code Online (Sandbox Code Playgroud)

这会产生错误:

TypeError: no supported conversion for types: (dtype('O'),)
Run Code Online (Sandbox Code Playgroud)

我将为此打开一个新查询.

Til*_*ill 5

你可以使用:

class_weight:{dict,'balanced'},可选

将类i的参数C设置class_weight[i]*C为SVC.如果没有给出,所有课程都应该有一个重量."平衡"模式使用y的值自动调整与输入数据中的类频率成反比的权重n_samples / (n_classes * np.bincount(y))

clf = OneVsRestClassifier(svm.SVC(kernel='rbf', class_weight='balanced'))
Run Code Online (Sandbox Code Playgroud)

资源