logistic回归

混淆矩阵以及热力图

import numpy as np 
import pandas as pd
from sklearn import linear_model
from sklearn import model_selection

sdata = pd.read_csv('../input/traindatas/char9/Run or Walk.csv')
#print(sdata.head())
prediction = sdata.columns[4:]
X = sdata.loc[:,prediction]
y = sdata.activity
x_train,x_test,y_train,y_test = model_selection.train_test_split(X,y,test_size=0.25,random_state = 1234)
sklearn_logistic = linear_model.LogisticRegression()
sklearn_logistic.fit(x_train,y_train)

from sklearn import metrics
cm = metrics.confusion_matrix(y_test,sklearn_predict,labels = [0,1])
print(cm)

import seaborn as sb
import matplotlib.pyplot as plt
sb.heatmap(cm,annot=True, fmt=".2f" ,cmap="YlGnBu")
plt.show()

直接结果