python—逻辑回归算法

简单介绍一个逻辑回归算法的例子
例子:通过计算自己的复习时长和复习效率,计算是否能考试985高校。

from sklearn.linear_model import LogisticRegression
# 复习情况,格式为[时长,效率]
x_train=[(0,0),(7,0.9),(15,0.4),(13,0.9),(15,0.4),(16,0.4),(16,0.8),(15,0.7),(17,0.2),(7.5,0.8),(17,0.9),(18,0.1),(18,0.6),(18,0.8)]
y_train=[0,0,0,1,0,0,1,1,0,1,1,0,1,1] #0考不上,1考得上
# 创建并训练逻辑回归模型
reg=LogisticRegression()
reg.fit(x_train,y_train)
# 测试模型
x_test=[(3,0.1),(8,0.3),(7,0.2),(4,0.2),(4,0.4)]
y_test=[0,0,0,0,0]
score=reg.score(x_test,y_test)
# 预测并输出预测结果
learning=[(16,0.9)]
result=reg.predict_proba(learning)
msg='''模型得分:{0}
复习时长为:{1[0]},效率为:{1[1]}
您考不上的概率为:{2[0]}
您考的上的概率为:{2[1]}
总和判断,您会:{3}'''.format(score,learning[0],result[0],'考不上'if result[0][0]>0.5 else "考得上")
print(msg)

输出结果如下:

模型得分:1.0
复习时长为:16,效率为:0.9
您考不上的概率为:0.3563081851629081
您考的上的概率为:0.6436918148370919
综合判断,您会:考得上

所以,好好学习,985不是梦。
本期分享就到这里,边学习边分享。

你可能感兴趣的:(逻辑回归算法,python,考985高校,算法,机器学习,python,深度学习,人工智能)