-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathdiabetesClassifiers.py
More file actions
35 lines (28 loc) · 1.09 KB
/
Copy pathdiabetesClassifiers.py
File metadata and controls
35 lines (28 loc) · 1.09 KB
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
27
28
29
30
31
32
33
34
35
#%%
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LogisticRegression
from sklearn.metrics import confusion_matrix
from sklearn.metrics import classification_report
import pandas as pd
import numpy as np
#%%
diabetesDF: pd.DataFrame = pd.read_csv('./data/diabetes.csv')
attributeDF: pd.DataFrame = diabetesDF.drop("Outcome", axis=1)
labels: pd.Series = diabetesDF["Outcome"]
#%%
attributeTraining, attributeTesting, labelTraining, labelTesting = train_test_split(attributeDF,labels)
# %%
logisticRegClassifier: LogisticRegression = LogisticRegression(solver='lbfgs', max_iter=200)
# %%
regModel: LogisticRegression = logisticRegClassifier.fit(attributeTraining,labelTraining)
# %%
print(regModel.score(attributeTesting,labelTesting))
# %%
predictedLabels: np.ndarray = regModel.predict(attributeTesting)
# %%
confusionMatrixDF: pd.DataFrame = pd.DataFrame(confusion_matrix(labelTesting,predictedLabels),index=['Actual +','Actual -'],columns=['Predicted +', 'Predicted -'])
# %%
print(confusionMatrixDF)
# %%
print(classification_report(labelTesting,predictedLabels))
# %%