Le précédent a capturé visuellement la différence entre les ** deux types de fonctions de pondération uniforme et distance ** dans la méthode * k-voisinage.
import numpy as np
import pandas as pd
#bibliothèques basées sur sklearn
from sklearn import datasets #base de données
from sklearn.model_selection import train_test_split #Répartition des données
from sklearn.neighbors import KNeighborsClassifier #Modèle de classification
from sklearn.neighbors import KNeighborsRegressor #Modèle de régression
#bibliothèques basées sur matplotlib
import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap
!pip install japanize-matplotlib #Module compatible avec l'affichage japonais
import japanize_matplotlib
#Obtenir l'ensemble de données
boston = datasets.load_boston()
#Extraire les variables explicatives et les variables objectives
X = boston.data[:, 5].reshape(len(boston.data), 1)
y = (boston.target).reshape(len(boston.target), 1)
#Division des données pour la formation / les tests
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state = 0)
#paramètre k
n_neighbors = 14
#Variable qui stocke le taux de précision
score = []
for w in ['uniform', 'distance']:
#Génération de modèle
model = KNeighborsRegressor(n_neighbors, weights=w)
model = model.fit(X_train, y_train)
#Taux de réponse correct des données d'entraînement
r_train = model.score(X_train, y_train)
score.append(r_train)
#Taux de réponse correct des données de test
r_test = model.score(X_test, y_test)
score.append(r_test)
#Représenté dans un bloc de données
score = np.array(score)
pd.DataFrame(score.reshape(2,2),
columns = ['train', 'test'],
index = ['uniform', 'distance'])
#paramètre k
n_neighbors = 14
#Génération d'instance
model_u = KNeighborsRegressor(n_neighbors, weights='uniform')
model_d = KNeighborsRegressor(n_neighbors, weights='distance')
#Génération de modèle
model_u = model_u.fit(X_train, y_train)
model_d = model_d.fit(X_train, y_train)
#Prévoir
y_u = model_u.predict(X_test)
y_d = model_d.predict(X_test)
plt.figure(figsize=(14,6))
#Nuage de points
plt.scatter(X_test, y_u, color='slateblue', lw=1, label='Valeur prédite(uniform)')
plt.scatter(X_test, y_d, color='tomato', lw=1, label='Valeur prédite(distance)')
plt.scatter(X_test, y_test, color='lightgrey', label='La valeur de mesure(test)')
plt.legend(fontsize=15)
plt.xlim(3, 9.5)
plt.show()
#Obtenir l'ensemble de données
iris = datasets.load_iris()
#Extraire uniquement les variables explicatives et les variables objectives
X = iris.data[:, :2]
y = iris.target
y = y.reshape(-1, 1) #Conversion de forme
#Définissez chaque variable après avoir extrait seulement 2 valeurs
data = np.hstack([X, y]) # X,Combinez y
data = data[data[:, 2] != 0] #Extraire seulement 2 valeurs
X = data[:, :2]
y = data[:, -1]
#Répartition des données
X_train, X_test, y_train, y_test = train_test_split(X, y, stratify = y, random_state = 0)
#paramètre k
n_neighbors = 15
#Variable qui stocke le taux de précision
score = []
for i, w in enumerate(['uniform', 'distance']):
#Génération de modèle
model = KNeighborsClassifier(n_neighbors, weights=w)
model = model.fit(X_train, y_train)
#Données d'entraînement
r_train = model.score(X_train, y_train)
score.append(r_train)
#données de test
r_test = model.score(X_test, y_test)
score.append(r_test)
#Représenté dans un bloc de données
score = np.array(score)
pd.DataFrame(score.reshape(2,2),
columns = ['train', 'test'],
index = ['uniform', 'distance'])
#paramètre k
n_neighbors = 15
#Espacement des mailles
h = 0.02
#Générer une carte de couleurs pour le mappage
cmap_surface = ListedColormap(['mistyrose', 'lightcyan'])
cmap_dot = ListedColormap(['tomato', 'slateblue'])
plt.figure(figsize=(18,6))
for j, w in enumerate(['uniform', 'distance']):
#Générer un modèle
model = KNeighborsClassifier(n_neighbors, weights = w)
model = model.fit(X_train, y_train)
#Définir les données de test
X, y = X_test, y_test
# x,Obtenez les valeurs minimale et maximale de l'axe des y
x_min, x_max = X[:, 0].min() - 1, X[:, 0].max() + 1
y_min, y_max = X[:, 1].min() - 1, X[:, 1].max() + 1
#Générer des colonnes de grille à des intervalles de maillage spécifiés
xx, yy = np.meshgrid(np.arange(x_min, x_max, h),
np.arange(y_min, y_max, h))
#Prédire en passant la séquence de grille au modèle
z = np.c_[xx.ravel(), yy.ravel()] #Aplatir à une dimension, puis joindre
Z = model.predict(z) #Prévoir
Z = Z.reshape(xx.shape) #Conversion de forme
#dessin
plt.subplot(1, 2, j + 1)
plt.pcolormesh(xx, yy, Z, cmap=cmap_surface) #Tracé de couleur
plt.scatter(X[:, 0], X[:, 1], c=y, cmap=cmap_dot, s=30)
plt.xlim(xx.min(), xx.max())
plt.ylim(yy.min(), yy.max())
plt.xlabel('La longueur de la pièce', fontsize=12)
plt.ylabel('Largeur de la pièce', fontsize=12)
plt.title("'%s'" % (w), fontsize=18)
plt.show()