diff --git a/newFeat.py b/newFeat.py index e97a53f..dd10c3f 100644 --- a/newFeat.py +++ b/newFeat.py @@ -1,3 +1,4 @@ +#Bhanu import pandas as pd import matplotlib.pyplot as plt import seaborn as sns @@ -23,9 +24,10 @@ import warnings warnings.filterwarnings('ignore') +# stratify the outcome outcome=diab['Outcome'] data=diab[diab.columns[:10]] -train,test=train_test_split(diab,test_size=0.25,random_state=0,stratify=diab['Outcome'])# stratify the outcome +train,test=train_test_split(diab,test_size=0.25,random_state=0,stratify=diab['Outcome']) train_X=train[train.columns[:10]] test_X=test[test.columns[:10]] train_Y=train['Outcome'] @@ -42,4 +44,4 @@ model = LogisticRegression() model.fit(train_X,train_Y) prediction=model.predict(test_X) -print('The accuracy of the Logistic Regression is',metrics.accuracy_score(prediction,test_Y)) \ No newline at end of file +print('The accuracy of the Logistic Regression is',metrics.accuracy_score(prediction,test_Y))