knn 算法实现皮玛印地安人糖尿病预测

KNN 算法是一种基于实例的学习算法，用于分类和回归。在本问题中，我们将使用 KNN 算法来预测皮玛印第安人是否患有糖尿病。

首先，我们需要收集数据集并将其分为训练集和测试集。我们可以使用 UCI Machine Learning Repository 中的皮玛印第安人糖尿病数据集。该数据集包含了768个样本，每个样本有8个特征（如怀孕次数、血糖浓度等），并且标签是0（没有糖尿病）或1（有糖尿病）。

接着，我们需要对数据进行预处理，包括特征缩放和特征选择等。然后，我们可以使用 KNN 算法来训练模型。KNN 算法通过计算测试样本与每个训练样本之间的距离（通常使用欧几里得距离），并找到与测试样本最近的 k 个训练样本。然后，它使用这些 k 个训练样本的标签来预测测试样本的标签。

最后，我们可以使用测试集来评估模型的性能。我们可以计算模型的准确率、精确率、召回率和 F1 分数等指标。

下面是 Python 代码实现：

import numpy as np
from sklearn.neighbors import KNeighborsClassifier
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score
from sklearn.preprocessing import StandardScaler
from sklearn.feature_selection import SelectKBest, f_classif

# Load dataset
data = np.loadtxt('pima-indians-diabetes.data.csv', delimiter=',')
X = data[:, :-1]
y = data[:, -1]

# Split dataset into training set and test set
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)

# Feature scaling
scaler = StandardScaler()
X_train = scaler.fit_transform(X_train)
X_test = scaler.transform(X_test)

# Feature selection
selector = SelectKBest(f_classif, k=4)
X_train = selector.fit_transform(X_train, y_train)
X_test = selector.transform(X_test)

# Train model
model = KNeighborsClassifier(n_neighbors=5)
model.fit(X_train, y_train)

# Predict
y_pred = model.predict(X_test)

# Evaluate
print('Accuracy:', accuracy_score(y_test, y_pred))
print('Precision:', precision_score(y_test, y_pred))
print('Recall:', recall_score(y_test, y_pred))
print('F1 score:', f1_score(y_test, y_pred))

这里我们使用了 sklearn 库中的 KNeighborsClassifier 类来实现 KNN 算法，并使用 train_test_split() 函数将数据集分为训练集和测试集。我们还使用了 StandardScaler 类来进行特征缩放，并使用 SelectKBest 类来进行特征选择。最后，我们使用准确率、精确率、召回率和 F1 分数等指标来评估模型的性能。