Untitled

 avatar
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