Objetivo do exemplo é predizer o valor de um imóvel em Boston utilizando dados já existentes referente a outros imóveis.
from sklearn.datasets import load_boston
import matplotlib.pyplot as plt
from sklearn.neighbors import KNeighborsRegressor
boston = load_boston()
#numero de vizinhos próximos
K = 9
#algoritmo para ser utilizado
knn = KNeighborsRegressor(n_neighbors=K)
#numero de registros utilizado para gerar o modelo
n_data = 100
x, y = boston.data[:n_data], boston.target[:n_data]
knn.fit(x, y)
valor_predicao = knn.predict([boston.data[0]])
valor_real= boston.target[0]
print("Valor Real: ",valor_real)
print("Valor Estimado: ",valor_predicao[0])
import numpy as np
#Predizendo N_data valores
y_predicted = knn.predict(x)
#Visualização gráfica
plt.plot(np.linspace(-1, 1, n_data), y, label='Valores Reais', color='black')
plt.plot(np.linspace(-1, 1, n_data), y_predicted, label='Valores Preditos', color='red')
plt.legend()
plt.show()