1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | from sklearn.datasets import load_iris iris = load_iris() from sklearn.model_selection import train_test_split # ÀԷ°ú Ãâ·ÂÀ» ¼³Á¤ÇÑ´Ù. X = iris.data y = iris.target # Àüü µ¥ÀÌÅ͸¦ ÇнÀ µ¥ÀÌÅÍ¿Í Å×½ºÆ® µ¥ÀÌÅÍ ºñÀ² (80:20)À¸·Î ºÐÇÒÇÑ´Ù. X_train,X_test,y_train,y_test = train_test_split(X,y,test_size=0.2,random_state=4) print(X_train.shape) from sklearn.neighbors import KNeighborsClassifier from sklearn import metrics # ÇнÀ ´Ü°è knn = KNeighborsClassifier(n_neighbors=6) knn.fit(X_train, y_train) # Å×½ºÆ® ´Ü°è y_pred = knn.predict(X_test) # Á¤È®µµ Á¡¼ö Ãâ·Â scores = metrics.accuracy_score(y_test, y_pred) print(scores) | cs |