如何使用ScikitLearn.jl和Gadfly.jl绘制SVM决策边界?

时间:2017-04-29 20:36:11

标签: plot julia gadfly

我想使用Gadfly绘制SVM模型的决策边界。我尝试了LIBSVM.jl,但它没有给出SVM模型的任何系数。 ScikitLearn.jl提供decision_function,这似乎有帮助。

我发现this example使用decision_functionPyPlot来绘制决策边界。我想知道如何使用Gadfly做到这一点?

在示例中,他们使用contourf中的PyPlot

x_min, x_max = minimum(X[:, 1]) - .5, maximum(X[:, 1]) + .5
y_min, y_max = minimum(X[:, 2]) - .5, maximum(X[:, 2]) + .5
xx, yy = meshgrid(x_min:h:x_max, y_min:h:y_max)

z = reshape(z, size(xx)...)
ax[:contourf](xx, yy, z, cmap=cm, alpha=.8)

我尝试在Geom.contour中使用Gadfly

plot(z=z, x=xx, y=yy, Geom.contour)

它引发了错误“Stat.contour要求z的长度为长度(x)的长度(y)”。

在我的示例中,zxxyy都是200×247阵列,它们在contourf中工作。我不确定如何将xxyy转换为200×1和247×​​1数组,以便使用Geom.contour

任何建议都表示赞赏。

0 个答案:

没有答案