Besoin d'aide avec l'algorithme KNN

W

witsman

Hé,
J'essaie d'appliquer la méthode K-Nearest Neighbors (KNN) à un ensemble de données en Python, mais je continue à recevoir une erreur que je ne peux pas dépasser, alors j'ai lu les principes fondamentaux de cet article. Quelqu'un pourrait-il m'aider s'il vous plaît?
Voici la section pertinente de mon code :
Code: 
import numpy as np
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.neighbors import KNeighborsClassifier
from sklearn.metrics import accuracy_score

# Custom dataset
X = np.array([[0, 1], [2, 3], [4, 5], [6, 7], [8, 9]])
y = np.array([0, 1, 0, 1, 0])

# Split the data into training and test sets
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

# Standardize the features
scaler = StandardScaler()
X_train = scaler.fit_transform(X_train)
X_test = scaler.transform(X_test)

# Implement KNN
k = 4
knn = KNeighborsClassifier(n_neighbors=k)
knn.fit(X_train, y_train)

# Make predictions on the test set
y_pred = knn.predict(X_test)

# Calculate accuracy
accuracy = accuracy_score(y_test, y_pred)
print(f"Accuracy: {accuracy}")
Je reçois le message d'erreur suivant :
Code: 
ValueError: setting an array element with a sequence.
Je ne sais pas pourquoi cette erreur se produit ni comment y remédier. Avez-vous des suggestions? Merci!
 
Haut