监督学习python代码示例
时间: 2023-11-20 15:57:55 浏览: 35
以下是一个简单的监督学习Python代码示例,用于预测鸢尾花的品种:
```python
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier
# 加载数据集
iris = load_iris()
# 将数据集分为训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(iris['data'], iris['target'], random_state=0)
# 创建KNN分类器
knn = KNeighborsClassifier(n_neighbors=1)
# 训练模型
knn.fit(X_train, y_train)
# 预测新数据
X_new = [[5, 2.9, 1, 0.2]]
prediction = knn.predict(X_new)
# 输出预测结果
print("Predicted target name: {}".format(iris['target_names'][prediction]))
```
该代码使用了scikit-learn库中的KNN分类器来预测鸢尾花的品种。首先,代码加载了鸢尾花数据集,然后将数据集分为训练集和测试集。接下来,代码创建了一个KNN分类器,并使用训练集对其进行训练。最后,代码使用训练好的模型对新数据进行预测,并输出预测结果。