导航：首页 > 互联网科技 >

怎么用JavaScript预测鸢尾花品种

发表于：2025-02-04 作者：千家信息网编辑

千家信息网最后更新 2025年02月04日，这篇文章主要介绍"怎么用JavaScript预测鸢尾花品种"，在日常操作中，相信很多人在怎么用JavaScript预测鸢尾花品种问题上存在疑惑，小编查阅了各式资料，整理出简单好用的操作方法，希望对大家

千家信息网最后更新 2025年02月04日怎么用JavaScript预测鸢尾花品种

这篇文章主要介绍"怎么用JavaScript预测鸢尾花品种"，在日常操作中，相信很多人在怎么用JavaScript预测鸢尾花品种问题上存在疑惑，小编查阅了各式资料，整理出简单好用的操作方法，希望对大家解答"怎么用JavaScript预测鸢尾花品种"的疑惑有所帮助！接下来，请跟着小编一起来学习吧！

导入需要用到的模块

import pandas as pd
from sklearn.svm import SVC
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score

读入数据

df = pd.read_csv(r"iris\YT-Django-Iris-App-3xj9B0qqps-master\iris.csv")

将数据拆分成训练集和测试集

x = ['sepal_length','sepal_width','petal_length','petal_width']

X = df[x]
y = df['classification']

X_train, X_test, Y_train, Y_test = train_test_split(X,y,test_size=0.2,random_state=1)

训练数据集合测试数据集的比例是8:2

训练模型并预测

model = SVC(gamma='auto')
model.fit(X_train,Y_train)
predictions = model.predict(X_test)

输入数据预测

iris = [1,1,1,1]
results = model.predict([iris])
print(results)

结果results是一个列表

输出模型准确性

print(accuracy_score(Y_test,predictions))

运行代码得到结果为 0.966666666667

保存模型

pd.to_pickle(model,r"new_model.pickle")

如果需要用这个模型可以直接读入

model = pd.read_pickle(r"new_model.pickle")

完整代码

import pandas as pd
from sklearn.svm import SVC
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
df = pd.read_csv(r"iris\YT-Django-Iris-App-3xj9B0qqps-master\iris.csv")
print(df.head())
x = ['sepal_length','sepal_width','petal_length','petal_width']
X = df[x]
y = df['classification']
X_train, X_test, Y_train, Y_test = train_test_split(X,y,test_size=0.2,random_state=1)
model = SVC(gamma='auto')
model.fit(X_train,Y_train)
predictions = model.predict(X_test)
print(accuracy_score(Y_test,predictions))
pd.to_pickle(model,r"new_model.pickle")
model = pd.read_pickle(r"new_model.pickle")
iris = [1,1,1,1]
results = model.predict([iris])
print(results)