Untitled
unknown
plain_text
2 years ago
644 B
10
Indexable
# Function to train and evaluate kNN classifier on scaled data
def train_and_eval_knn_scaled(x_train,y_train,x_test,y_test,k_values):
scaler=StandardScaler()
x_train_scaled = scaler.fit_transform(x_train)
x_test_scaled = scaler.transform(x_test)
for k in k_values:
knn_classifier = KNeighborsClassifier(n_neighbors=k)
knn_classifier.fit(x_train_scaled, y_train)
y_pred_knn = knn_classifier.predict(x_test_scaled)
print(f'\nEvaluation for k={k}:')
eval_classifier(y_test, y_pred_knn)
k_values = list(range(1,11))
train_and_eval_knn_scaled(x_train,y_train,x_test,y_test,k_values) Editor is loading...
Leave a Comment