KNN, and Logistic Regression

ML_Models.py

### Here we load the train and test data for you to use.
(train_data, train_labels) = get_train_data(flatten = True)
(test_data, test_labels) = get_test_data(flatten = True)
############

knn  = KNeighborsClassifier(n_neighbors=10)
log_reg  = LogisticRegression()


knn.fit(train_data, train_labels)
log_reg.fit(train_data, train_labels)


predictions = knn.predict(test_data)
score = accuracy_score(test_labels, predictions)
print("KNN accuracy:",score)

predictions = log_reg.predict(test_data)
score = accuracy_score(test_labels, predictions)
print("Logistic Regression accuracy:",score)