knn算法pycharm实现
时间: 2024-01-04 08:20:38 浏览: 135
机器学习-KNN算法实现
KNN算法的实现可以使用Python编程语言在PyCharm中完成。下面是一个简单的KNN算法的实现示例:
```python
import numpy as np
from sklearn.neighbors import KNeighborsClassifier
# 创建训练数据集
X_train = np.array([[1, 2], [1, 4], [2, 2], [2, 3], [3, 2], [3, 4]])
y_train = np.array([0, 0, 0, 1, 1, 1])
# 创建KNN分类器对象
knn = KNeighborsClassifier(n_neighbors=3)
# 训练模型
knn.fit(X_train, y_train)
# 创建测试数据
X_test = np.array([[2, 1], [4, 3]])
# 预测测试数据的类别
y_pred = knn.predict(X_test)
# 打印预测结果
print("预测结果:", y_pred)
```
在这个示例中,我们使用了`numpy`库来创建训练数据集和测试数据集。然后,我们使用`sklearn.neighbors`模块中的`KNeighborsClassifier`类来创建KNN分类器对象。通过调用`fit`方法,我们将训练数据集传递给分类器进行训练。接下来,我们创建了一个测试数据集,并使用`predict`方法对测试数据进行预测。最后,我们打印出预测结果。
阅读全文