基于sklearn的K-Nearest Neighbor算法实现
编程环境
· python 3.7.7
· scikit-learn 0.22.1
· pandas 1.0.3
· numpy 1.18.1
· Graphviz 2.40.1
· jupyternotebook
基于鸢尾花数据的K近邻算法实现
- 初始设置,导入相关包
from IPython.display import set_matplotlib_formats, display
import pandas as pd # 导入pandas
import numpy as np # 导入numpy
import matplotlib.pyplot as plt
%matplotlib inline # 设置生成的图像直接在jupyter notebook的console中显示
- 分类鸢尾花数据
from sklearn.datasets import load_iris
iris_dataset = load_iris() # 调用load_iris方法读取数据进入irsi_dataset变量
鸢尾花数据为