algorithm/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-07-06 10:31:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

psf/black code formatting (#1277)

Browse Source
This commit is contained in:
William Zhang
2019-10-05 01:14:13 -04:00
committed by Christian Clauss
parent 07f04a2e55
commit 9eac17a408
291 changed files with 6014 additions and 4571 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
machine_learning/knn_sklearn.py
View File

@ -2,27 +2,30 @@ from sklearn.model_selection import train_test_split
from sklearn.datasets import load_iris
from sklearn.neighbors import KNeighborsClassifier
#Load iris file
# Load iris file
iris = load_iris()
iris.keys()
print('Target names: \n {} '.format(iris.target_names))
print('\n Features: \n {}'.format(iris.feature_names))
print("Target names: \n {} ".format(iris.target_names))
print("\n Features: \n {}".format(iris.feature_names))
#Train set e Test set
X_train, X_test, y_train, y_test = train_test_split(iris['data'],iris['target'], random_state=4)
# Train set e Test set
X_train, X_test, y_train, y_test = train_test_split(
iris["data"], iris["target"], random_state=4
)
#KNN
# KNN
knn = KNeighborsClassifier (n_neighbors = 1)
knn = KNeighborsClassifier(n_neighbors=1)
knn.fit(X_train, y_train)
#new array to test
X_new = [[1,2,1,4],
[2,3,4,5]]
# new array to test
X_new = [[1, 2, 1, 4], [2, 3, 4, 5]]
prediction = knn.predict(X_new)
print('\nNew array: \n {}'
'\n\nTarget Names Prediction: \n {}'.format(X_new, iris['target_names'][prediction]))
print(
"\nNew array: \n {}"
"\n\nTarget Names Prediction: \n {}".format(X_new, iris["target_names"][prediction])
)