Como desenhar os hiperplanos para SVM One-Versus-All?
Eu estava tentando desenhar os hiperplanos quando o SVM-OVA foi realizado da seguinte forma:
import matplotlib.pyplot as plt
import numpy as np
from sklearn.svm import SVC
x = np.array([[1,1.1],[1,2],[2,1]])
y = np.array([0,100,250])
classifier = OneVsRestClassifier(SVC(kernel='linear'))
Com base na resposta a esta pergunta Plot hyperplane Linear SVM python , escrevi o seguinte código:
fig, ax = plt.subplots()
# create a mesh to plot in
x_min, x_max = x[:, 0].min() - 1, x[:, 0].max() + 1
y_min, y_max = x[:, 1].min() - 1, x[:, 1].max() + 1
xx2, yy2 = np.meshgrid(np.arange(x_min, x_max, .2),np.arange(y_min, y_max, .2))
Z = classifier.predict(np.c_[xx2.ravel(), yy2.ravel()])
Z = Z.reshape(xx2.shape)
ax.contourf(xx2, yy2, Z, cmap=plt.cm.winter, alpha=0.3)
ax.scatter(x[:, 0], x[:, 1], c=y, cmap=plt.cm.winter, s=25)
# First line: class1 vs (class2 U class3)
w = classifier.coef_[0]
a = -w[0] / w[1]
xx = np.linspace(-5, 5)
yy = a * xx - (classifier.intercept_[0]) / w[1]
ax.plot(xx,yy)
# Second line: class2 vs (class1 U class3)
w = classifier.coef_[1]
a = -w[0] / w[1]
xx = np.linspace(-5, 5)
yy = a * xx - (classifier.intercept_[1]) / w[1]
ax.plot(xx,yy)
# Third line: class 3 vs (class2 U class1)
w = classifier.coef_[2]
a = -w[0] / w[1]
xx = np.linspace(-5, 5)
yy = a * xx - (classifier.intercept_[2]) / w[1]
ax.plot(xx,yy)
No entanto, foi isso que obtive:

As linhas estão claramente erradas: na verdade, os coeficientes angulares parecem corretos, mas não os interceptos. Em particular, a linha laranja estaria correta se convertida em 0,5 para baixo, a verde se convertida em 0,5 para a esquerda e a azul se convertida em 1,5 para cima.
Estou errado ao traçar as linhas ou o classificador não funciona corretamente por causa dos poucos pontos de treinamento?
Respostas
O problema é que o C
parâmetro de SVC
é muito pequeno (por padrão 1.0
). De acordo com esta postagem ,
Por outro lado, um valor muito pequeno de C fará com que o otimizador procure um hiperplano de separação de margem maior, mesmo que esse hiperplano classifique incorretamente mais pontos.
Portanto, a solução é usar um muito maior C
, por exemplo1e5
import matplotlib.pyplot as plt
import numpy as np
from sklearn.svm import SVC
from sklearn.multiclass import OneVsRestClassifier
x = np.array([[1,1.1],[1,2],[2,1]])
y = np.array([0,100,250])
classifier = OneVsRestClassifier(SVC(C=1e5,kernel='linear'))
classifier.fit(x,y)
fig, ax = plt.subplots()
# create a mesh to plot in
x_min, x_max = x[:, 0].min() - 1, x[:, 0].max() + 1
y_min, y_max = x[:, 1].min() - 1, x[:, 1].max() + 1
xx2, yy2 = np.meshgrid(np.arange(x_min, x_max, .2),np.arange(y_min, y_max, .2))
Z = classifier.predict(np.c_[xx2.ravel(), yy2.ravel()])
Z = Z.reshape(xx2.shape)
ax.contourf(xx2, yy2, Z, cmap=plt.cm.winter, alpha=0.3)
ax.scatter(x[:, 0], x[:, 1], c=y, cmap=plt.cm.winter, s=25)
def reconstruct(w,b):
k = - w[0] / w[1]
b = - b[0] / w[1]
if k >= 0:
x0 = max((y_min-b)/k,x_min)
x1 = min((y_max-b)/k,x_max)
else:
x0 = max((y_max-b)/k,x_min)
x1 = min((y_min-b)/k,x_max)
if np.abs(x0) == np.inf: x0 = x_min
if np.abs(x1) == np.inf: x1 = x_max
xx = np.linspace(x0,x1)
yy = k*xx+b
return xx,yy
xx,yy = reconstruct(classifier.coef_[0],classifier.intercept_[0])
ax.plot(xx,yy,'r')
xx,yy = reconstruct(classifier.coef_[1],classifier.intercept_[1])
ax.plot(xx,yy,'g')
xx,yy = reconstruct(classifier.coef_[2],classifier.intercept_[2])
ax.plot(xx,yy,'b')
Desta vez, porque um muito maior C
é adotado, o resultado parece melhor
