绘制SVC决策区域

Question

我正在使用moon_dataset在一本书中遵循一些SVC代码。

以下是代码：

import pandas as pd
import matplotlib.pyplot as plt
%matplotlib inline
from sklearn.datasets import make_moons
from sklearn.pipeline import Pipeline
from sklearn.preprocessing import PolynomialFeatures

X, y = make_moons(n_samples=100, noise=0.15)

rbf_kernel_svm_clf = Pipeline([
        ("scaler", StandardScaler()),
        ("svm_clf", SVC(kernel="rbf", gamma=5, C=0.001))
        ])
rbf_kernel_svm_clf.fit(X, y)

我是否尝试过用下面的代码绘制这些图中的任何一个，但到目前为止还没有。

plt.scatter(X, y)

有什么帮助吗？谢谢

Answer 1

要绘制决策区域，您需要的不仅仅是散点图。其中一个非常有用的模块是MLxtend，它使得用plot_decision_regions绘制拟合模型的决策区域变得非常容易。下面是如何使用您的示例完成它：

from mlxtend.plotting import plot_decision_regions

plt.figure(figsize=(12,8))
plot_decision_regions(X, y, clf=rbf_kernel_svm_clf.named_steps['svm_clf'], legend=2)

​

​