knn算法java实现鸢尾花,使用knn算法对鸢尾花数据进行分类

导入三剑客以及knn算法所用到的包

import pandas as pd

import numpy as np

import matplotlib.pyplot as plt

from sklearn.neighbors import KNeighborsClassifier

from sklearn.model_selection import  train_test_split

from sklearn.datasets import load_iris

# 创建鸢尾花对象

iris = load_iris()

# 获取鸢尾花数据

data = iris.data

# 获取鸢尾花的标签

target = iris.target

# 自动生成训练集和测试集数据

X_train, X_test, y_train, y_test = train_test_split(data, target, test_size=0.2)

# 创建knn对象

knn = KNeighborsClassifier()

# 训练数据

knn.fit(X_train, y_train)

# 预测

knn.predict(X_test)

# 获取得分

knn.score(X_test, y_test)

得分结果如下：