Python机器学习——逻辑回归案例(三)

引入披萨问题

分类问题:二分类

依旧还是披萨那些训练数据,只是最后判断是否好吃
在这里插入图片描述

1.通过分析已经确定属于逻辑回归问题

① 导入逻辑回归模型的库

# 导入需要的库
from sklearn import linear_model

② 四步走:导入训练数据 + 构建模型 + 训练模型 + 评估模型 + 模型预测

综合①②完整代码为:

from sklearn import linear_model

# 1.准备训练数据
X_train = [[8,11],[9,8.5],[11,15],[16,18],[12,11],[15,12]]
y_train = [1,0,1,0,1,0]

# 2.构建模型
model = linear_model.LogisticRegression()

# 3.训练模型
model.fit(X_train,y_train)

# 4.评估模型
y_pred = model.predict(X_train)
print(y_pred)  # [1 0 1 1 0 0]  与y_train相比,有4个相同
score = model.score(X_train,y_train)
print(score) # 0.6666666666666666

# 另一种模型评估的方法
# accuracy_score 精确度
# from sklearn.metrics import accuracy_score
# score = accuracy_score(y_train,y_pred)
# print(score) # 0.6666666666666666

# 5.模型预测
y_test = [[18,5]]
result = model.predict(y_test)
print("预测的结果为",result)
# 预测的结果为 [0]

细节解析:使用 accuracy_score(y的真实值,y的预测值) 进行模型评估时,必须导入库
from sklearn.metrics import accuracy_score

2.总结:

       ① 根据题意分析属于逻辑回归问题
       ② 导入逻辑回归模型库
       ③ 五步走:导入训练数据 + 构建模型 + 训练模型 + 评估模型 + 模型预测

发布了27 篇原创文章 · 获赞 7 · 访问量 2116

猜你喜欢

转载自blog.csdn.net/lucky_shi/article/details/105201492
今日推荐