Untitled
unknown
plain_text
a year ago
644 B
7
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