机器学习之使用逻辑回归算法预测学生录取情况案例

代码如下：

import numpy as np
datas = np.loadtxt('data1.txt',delimiter=',')
print(datas)
print(type(datas),datas.shape)
#取得样本特征数据和样本标签数据，并拆分成训练集和测试集
x_data = datas[:,0:2] #获取0列到2列
y_data = datas[:,2]   #单列索引（获取索引为2的列）
from sklearn.model_selection import train_test_split
x_train,x_test,y_train,y_test = train_test_split(x_data,y_data,test_size=0.2,random_state=9)
#建立逻辑回归模型
from sklearn.linear_model import LogisticRegression
model = LogisticRegression()
#训练模型
model.fit(x_train,y_train)
#模型测试
scores = model.score(x_test,y_test)
print(scores)
import matplotlib.pyplot as plt
def plot_decision_boundary(model,axis):
    x,y = np.meshgrid(np.linspace(axis[0],axis[1],1000),
                      np.linspace(axis[2],axis[3],1000))
    x_new = np.c_[x.ravel(),y.ravel()]
    y_pred = model.predict(x_new)
    zz = y_pred.reshape(x.shape)
    plt.contourf(x,y,zz)
    plt.show()
plot_decision_boundary(model,[0,100,0,100])

结果如下：