当前位置：首页 > news >正文

【机器学习算法基础】（基础机器学习课程）-10-逻辑回归-笔记

news 2025/7/16 10:57:03

一、模型的保存与加载

逻辑回归是一种常见的机器学习算法，广泛用于分类问题。为了在不同的时间或环境下使用训练好的模型，我们通常需要将其保存和加载。

保存模型

训练模型：首先，你需要用你的数据训练一个逻辑回归模型。例如，在Python中使用scikit-learn库，你可以这样做：
```
from sklearn.linear_model import LogisticRegression 
model = LogisticRegression() 
model.fit(X_train, y_train)
```
保存模型：训练好模型后，你需要将其保存到一个文件中。这样你就不需要每次都重新训练模型。常用的方法是使用Python的pickle库或joblib库。以下是使用joblib库的示例：
```
import joblib
joblib.dump(model, 'logistic_regression_model.pkl')
```
这里的'logistic_regression_model.pkl'是你保存模型的文件名。你可以选择任何合适的文件名和路径。

加载模型

加载模型：当你需要使用之前保存的模型时，可以将其从文件中加载回来。继续使用joblib库，加载模型的方法如下：
```
model = joblib.load('logistic_regression_model.pkl')
```
使用模型进行预测：加载模型后，你就可以直接使用它进行预测，而不需要重新训练模型。例如：

使用模型进行预测：加载模型后，你就可以直接使用它进行预测，而不需要重新训练模型。例如：

    predictions = model.predict(X_test)

总结

为什么保存模型？ 保存模型的主要目的是为了节省时间和计算资源。训练模型可能需要大量的数据和时间，而保存模型后可以在需要时直接加载和使用。
保存模型的方法：常用的库包括pickle和joblib，joblib在处理大文件时效率更高。
加载模型：加载模型非常简单，只需一行代码就可以将保存的模型加载回来，方便快速进行预测或分析。

实际应用场景

想象你是一个数据科学家，在一个项目中你花了几个小时甚至几天时间来训练一个逻辑回归模型，用于预测某些客户是否会购买你的产品。模型训练好了，你不希望每次运行预测时都重新训练模型，所以你会将其保存。当你需要在下个月或明年再次运行预测时，你只需要加载保存好的模型，而无需重新训练，大大节省了时间和资源。

这种方法不仅适用于逻辑回归模型，还适用于其他机器学习模型，如决策树、支持向量机和神经网络等。模型的保存与加载是机器学习工作流程中非常重要的一部分，可以提高效率和便利性。

二、逻辑回归定义，损失函数，优化与线性回归比较

逻辑回归是一种用于分类问题的机器学习算法，尽管名字中有“回归”二字，但它实际上用于二分类或多分类任务。

小结

逻辑回归用于分类问题，通过逻辑函数将线性模型的输出映射到概率值。
损失函数为对数损失函数，最小化它以优化模型。
优化方法主要为梯度下降。
与线性回归的比较显示了它们在任务、输出、假设、模型和损失函数上的不同之处。

三、逻辑回归预测癌症案例

场景设定

假设我们有一组数据，记录了若干患者的健康状况及其是否患有癌症。数据包括患者的年龄、体重、是否吸烟等信息，以及是否患有癌症（0表示未患癌，1表示患癌）。

目标

我们的目标是利用这些数据训练一个逻辑回归模型，当我们输入新的患者信息时，模型可以预测该患者是否患有癌症的概率。

步骤

收集和准备数据：

数据集可能包含以下信息：

| 年龄 | 体重 | 吸烟 | 是否患癌 |
|----|----|----|-------|
| 45 | 70 | 1  | 1     |
| 50 | 65 | 0  | 0     |
| 40 | 80 | 1  | 1     |
| 30 | 60 | 0  | 0     |

每一行代表一个患者的信息。

训练模型：

使用这些数据来训练逻辑回归模型。

from sklearn.linear_model import LogisticRegression
import pandas as pd# 假设数据已经被加载到一个DataFrame中
data = pd.DataFrame({'age': [45, 50, 40, 30],'weight': [70, 65, 80, 60],'smoke': [1, 0, 1, 0],'cancer': [1, 0, 1, 0]
})X = data[['age', 'weight', 'smoke']]
y = data['cancer']model = LogisticRegression()
model.fit(X, y)

保存模型：
- 训练好模型后，将其保存到文件中，以便后续使用。
```
import joblib
joblib.dump(model, 'cancer_prediction_model.pkl')
```
加载模型和预测：

当有新的患者信息时，可以加载保存的模型并进行预测。

# 加载模型
model = joblib.load('cancer_prediction_model.pkl')# 新患者的信息
new_patient = [[50, 70, 1]]# 预测该患者患癌的概率
cancer_probability = model.predict_proba(new_patient)[0][1]
print(f'该患者患癌的概率是: {cancer_probability:.2f}')