本文通过逻辑回归分类来预测一个学生是否被录取,来熟悉python逻辑回归模型及模型的评估,数据集有需要的可以联系我qq:1344184686。
一、读入数据
#1、【读入数据】
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
#gpa为绩点,gre为一种英语等级考试
data = pd.read_csv('admissions.csv')
data.head()
读入结果如下所示:
熟悉一下sigmoid函数:
#2、【sigmoid函数】
def sigmoid(z):
h = 1/(1+np.exp(-z))
return h
z = np.linspace(-5,5,40)
h = sigmoid(z)
plt.plot(z,h)
sigmoid函数图像如下图:
二、选择模型,训练并预测
#3、【用验证集测试】
from sklearn.linear_model import LogisticRegre