HI
Using KNN classifier in Python we can do classification of data.
# 1 ---------------------------------
https://github.com/susanli2016/Machine-Learning-with-Python/blob/master/diabetes.csv
# 2 ---------------------------------
https://www.kaggle.com/code/mragpavank/pima-indians-diabetes-database/input
# 3---------------------------------
from google.colab import drive
drive.mount('/content/drive')
# 4---------------------------------
cd /content/drive/MyDrive/MLProject
# 5---------------------------------
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
diabetesdata = pd.read_csv('diabetesdata.csv')
print(diabetesdata.columns)
# 6---------------------------------
diabetesdata.head()
print("dimension of diabetesdata data: {}".format(diabetesdata.shape))
print(diabetesdata.groupby('Outcome').size())
diabetesdata.info()
# 7 ---------------------------------
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(diabetesdata.loc[:, diabetesdata.columns != 'Outcome'], diabetesdata['Outcome'], stratify=diabetesdata['Outcome'], random_state=66)
# 8---------------------------------
from sklearn.neighbors import KNeighborsClassifier
# 9---------------------------------
knn = KNeighborsClassifier(n_neighbors=9)
knn.fit(X_train, y_train)
#10 Save the trained model---------------------------------
import joblib
joblib.dump(knn, "knn_model.pkl")
# 11 Load the model---------------------------------
import joblib
knn_loaded = joblib.load("knn_model.pkl")
inputdata=[[10,140,80,0,0,27.1,1.440,55]]
print(knn_loaded.predict(inputdata))
No comments:
Post a Comment