人工智能概论作业三

import numpy as np

import matplotlib.pyplot as plt

from sklearn.datasets import make_classification

from sklearn.model_selection import train_test_split

 

X, y = make_classification(n_samples=1000, n_features=2, n_informative=2,

                                                      n_redundant=0, n_clusters_per_class=1, class_sep=0.5,

                                                      random_state=42)

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

 

from sklearn.linear_model import LogisticRegression

from sklearn.tree import DecisionTreeClassifier

from sklearn.svm import SVC

 

classifiers = {

        "Logistic Regression": LogisticRegression(random_state=42),

        "Decision Tree": DecisionTreeClassifier(random_state=42),

        "SVM": SVC(kernel="linear", C=0.025, random_state=42),

}

 

for name, clf in classifiers.items():

        clf.fit(X_train, y_train)

        xx, yy = np.meshgrid(np.linspace(-3, 3, 500), np.linspace(-3, 3, 500))

        Z = clf.predict(np.c_[xx.ravel(), yy.ravel()])

        Z = Z.reshape(xx.shape)

        plt.contourf(xx, yy, Z, cmap=plt.cm.RdBu, alpha=0.5)

        plt.scatter(X_train[:, 0], X_train[:, 1], c=y_train, cmap=plt.cm.RdBu, edgecolors='k')

        plt.title(name)

        plt.show()

 

fig, axs = plt.subplots(nrows=1, ncols=len(classifiers), figsize=(15, 5))

 

for ax, (name, clf) in zip(axs, classifiers.items()):

        clf.fit(X_train, y_train)

        xx, yy = np.meshgrid(np.linspace(-3, 3, 500), np.linspace(-3, 3, 500))

        Z = clf.predict(np.c_[xx.ravel(), yy.ravel()])

        Z = Z.reshape(xx.shape)

        ax.contourf(xx, yy, Z, cmap=plt.cm.RdBu, alpha=0.5)

        ax.scatter(X_train[:, 0], X_train[:, 1], c=y_train, cmap=plt.cm.RdBu, edgecolors='k')

        ax.set_title(name)

 

plt.show()