Scikit-learn (`sklearn`) 教程

embedded/2024/9/22 22:34:07/

sklearn__0">Scikit-learn (sklearn) 教程

Scikit-learn 是 Python 中最流行的机器学习库之一，提供了丰富的机器学习算法、数据预处理工具以及模型评估方法，广泛应用于分类、回归、聚类和降维等任务。

在本教程中，我们将介绍如何使用 Scikit-learn 进行数据加载、特征处理、模型训练与评估，并展示一些常用的机器学习模型。

1. 安装 Scikit-learn

你可以使用以下命令安装 scikit-learn：

pip install scikit-learn

2. Scikit-learn 的核心组件

  • 数据集：提供内置数据集和数据集加载工具。
  • 特征工程：包括特征缩放、编码、缺失值处理等。
  • 模型：提供分类、回归、聚类、降维等多种算法。
  • 模型评估：包括交叉验证、网格搜索等。

3. 数据集

Scikit-learn 提供了多种内置数据集（例如 Iris、Boston），并且提供了用于加载外部数据集的工具。

3.1 加载内置数据集

例如，加载 Iris 数据集：

python">from sklearn.datasets import load_iris# 加载 Iris 数据集
iris = load_iris()
print(iris.feature_names)  # 输出特征名称
print(iris.target_names)   # 输出目标类别名称# 特征数据
X = iris.data
# 目标数据
y = iris.targetprint(X.shape, y.shape)
3.2 使用 Pandas 加载 CSV 数据

你也可以使用 Pandas 加载本地 CSV 数据：

python">import pandas as pd# 加载 CSV 数据
data = pd.read_csv('data.csv')# 分离特征和目标
X = data.drop('target_column', axis=1)
y = data['target_column']

4. 数据预处理

Scikit-learn 提供了一些常用的特征预处理工具，例如标准化、归一化、标签编码等。

4.1 标准化与归一化
  • 标准化：将数据转换为均值为 0，方差为 1 的正态分布。
  • 归一化：将数据缩放到 [0, 1] 或 [-1, 1] 的范围。
python">from sklearn.preprocessing import StandardScaler, MinMaxScaler# 标准化
scaler = StandardScaler()
X_scaled = scaler.fit_transform(X)# 归一化
min_max_scaler = MinMaxScaler()
X_normalized = min_max_scaler.fit_transform(X)
4.2 标签编码

将分类标签转换为数字编码：

python">from sklearn.preprocessing import LabelEncoder# 标签编码
label_encoder = LabelEncoder()
y_encoded = label_encoder.fit_transform(y)

5. 训练/测试集拆分

在进行模型训练前，通常需要将数据集划分为训练集和测试集。Scikit-learn 提供了 train_test_split 函数来完成这一操作。

python">from sklearn.model_selection import train_test_split# 划分训练集和测试集，比例为 80:20
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)print(X_train.shape, X_test.shape)

6. 模型训练

Scikit-learn 提供了丰富的机器学习算法。常见的分类、回归和聚类算法都可以通过 fit() 方法来训练模型。

6.1 分类任务示例：K 最近邻 (KNN)

K 最近邻算法是一种经典的分类算法。以下是使用 KNN 进行分类的示例：

python">from sklearn.neighbors import KNeighborsClassifier# 创建 KNN 模型
knn = KNeighborsClassifier(n_neighbors=3)# 训练模型
knn.fit(X_train, y_train)# 在测试集上进行预测
y_pred = knn.predict(X_test)
6.2 回归任务示例：线性回归

对于回归任务，可以使用线性回归模型进行训练：

python">from sklearn.linear_model import LinearRegression# 创建线性回归模型
lr = LinearRegression()# 训练模型
lr.fit(X_train, y_train)# 在测试集上进行预测
y_pred = lr.predict(X_test)

7. 模型评估

Scikit-learn 提供了多种评估指标，用于衡量模型的性能。

7.1 分类模型评估
  • 准确率：分类模型中常用的评估指标，表示预测正确的样本占总样本的比例。
python">from sklearn.metrics import accuracy_score, classification_report, confusion_matrix# 计算准确率
accuracy = accuracy_score(y_test, y_pred)
print(f'Accuracy: {accuracy}')# 分类报告
print(classification_report(y_test, y_pred))# 混淆矩阵
print(confusion_matrix(y_test, y_pred))
7.2 回归模型评估
  • 均方误差：常用的回归模型评估指标，衡量预测值与真实值的差距。
python">from sklearn.metrics import mean_squared_error, r2_score# 计算均方误差
mse = mean_squared_error(y_test, y_pred)
print(f'Mean Squared Error: {mse}')# R^2 分数
r2 = r2_score(y_test, y_pred)
print(f'R^2 Score: {r2}')

8. 交叉验证

交叉验证是一种常用的模型评估方法，可以更稳健地评估模型性能。Scikit-learn 提供了 cross_val_score 来实现交叉验证。

python">from sklearn.model_selection import cross_val_score# 使用交叉验证评估模型性能，使用 5 折交叉验证
scores = cross_val_score(knn, X, y, cv=5)
print(f'Cross-validation scores: {scores}')
print(f'Average score: {scores.mean()}')

9. 模型调参

在实际应用中，找到最优的超参数组合非常重要。Scikit-learn 提供了 GridSearchCV 和 RandomizedSearchCV 来进行超参数调优。

9.1 网格搜索（Grid Search）
python">from sklearn.model_selection import GridSearchCV# 定义参数网格
param_grid = {'n_neighbors': [3, 5, 7],'weights': ['uniform', 'distance']
}# 进行网格搜索
grid_search = GridSearchCV(KNeighborsClassifier(), param_grid, cv=5)
grid_search.fit(X_train, y_train)# 输出最佳参数
print(f'Best parameters: {grid_search.best_params_}')
9.2 随机搜索（Randomized Search）
python">from sklearn.model_selection import RandomizedSearchCV# 定义随机搜索参数
param_distributions = {'n_neighbors': [3, 5, 7],'weights': ['uniform', 'distance']
}# 随机搜索
random_search = RandomizedSearchCV(KNeighborsClassifier(), param_distributions, cv=5, n_iter=10)
random_search.fit(X_train, y_train)# 输出最佳参数
print(f'Best parameters: {random_search.best_params_}')

10. 管道 (Pipeline)

Pipeline 是 scikit-learn 中的一个非常有用的工具，它将多个步骤组合在一起，形成一个工作流。通过 Pipeline，我们可以将数据预处理和模型训练整合为一个过程，方便进行交叉验证和超参数调优。

10.1 创建管道
python">from sklearn.pipeline import Pipeline
from sklearn.preprocessing import StandardScaler
from sklearn.svm import SVC# 创建一个包含标准化和 SVM 分类器的管道
pipeline = Pipeline([('scaler', StandardScaler()),  # 数据标准化('svc', SVC())                 # 支持向量机分类器
])# 使用管道进行训练和预测
pipeline.fit(X_train, y_train)
y_pred = pipeline.predict(X_test)# 评估模型
print(f'Accuracy: {accuracy_score(y_test, y_pred)}')
10.2 在管道中使用网格搜索

你可以在 Pipeline 中使用超参数调优，调整管道中的每个步骤的参数。

python">from sklearn.model_selection import GridSearchCV# 定义参数网格
param_grid = {'svc__C': [0.1, 1, 10],'svc__kernel': ['linear', 'rbf']
}# 使用管道进行网格搜索
grid_search = GridSearchCV(pipeline, param_grid, cv=5)
grid_search.fit(X_train, y_train)# 输出最佳参数
print(f'Best parameters: {grid_search.best_params_}')# 使用最佳模型进行预测
best_pipeline = grid_search.best_estimator_
y_pred = best_pipeline.predict(X_test)# 评估结果
print(f'Accuracy: {accuracy_score(y_test, y_pred)}')

11. 特征选择

在机器学习中，特征选择是非常重要的一步。通过去除无用或冗余的特征，可以提升模型的性能。scikit-learn 提供了多种特征选择的方法。

11.1 使用 SelectKBest 进行特征选择

SelectKBest 是一种常见的特征选择方法，它根据某种评分标准（如 f_classif）选择前 K 个最重要的特征。

python">from sklearn.feature_selection import SelectKBest, f_classif# 使用 SelectKBest 选择前 2 个最重要的特征
selector = SelectKBest(f_classif, k=2)
X_new = selector.fit_transform(X_train, y_train)# 打印被选择的特征
print(X_new.shape)
11.2 在管道中使用特征选择

你可以将特征选择步骤集成到 Pipeline 中，以便与其他步骤（如标准化和模型训练）一起进行处理。

python">from sklearn.feature_selection import SelectKBest
from sklearn.preprocessing import StandardScaler
from sklearn.svm import SVC
from sklearn.pipeline import Pipeline# 创建一个包含特征选择、标准化和 SVM 分类器的管道
pipeline = Pipeline([('select', SelectKBest(f_classif, k=2)),  # 特征选择('scaler', StandardScaler()),             # 标准化('svc', SVC())                            # 支持向量机
])# 训练模型
pipeline.fit(X_train, y_train)# 预测结果
y_pred = pipeline.predict(X_test)
print(f'Accuracy: {accuracy_score(y_test, y_pred)}')

12. 聚类

scikit-learn 提供了多种聚类算法。聚类是无监督学习中的一种任务，目标是将数据划分为多个组（簇），其中同一簇的对象相似度较高。

12.1 K-means 聚类

K-means 是一种经典的聚类算法，它通过最小化簇内的方差将数据划分为 K 个簇。

python">from sklearn.cluster import KMeans# 创建 K-means 模型，指定 3 个簇
kmeans = KMeans(n_clusters=3, random_state=42)# 训练模型
kmeans.fit(X)# 预测簇标签
y_kmeans = kmeans.predict(X)# 打印每个样本所属的簇
print(y_kmeans)
12.2 层次聚类 (Agglomerative Clustering)

层次聚类通过不断合并最近的簇来构建层次树结构。你可以指定合并停止的簇数量。

python">from sklearn.cluster import AgglomerativeClustering# 创建层次聚类模型
agg_clustering = AgglomerativeClustering(n_clusters=3)# 训练模型
y_agg = agg_clustering.fit_predict(X)# 打印簇标签
print(y_agg)

13. 降维

降维技术用于将高维数据映射到低维空间，减少维度，同时尽可能保留原始数据的信息量。常见的降维方法有主成分分析（PCA）和线性判别分析（LDA）。

13.1 PCA 降维

主成分分析（PCA）是一种线性降维技术，找到数据的主要方向，最大限度地保留数据的方差。

python">from sklearn.decomposition import PCA# 创建 PCA 模型，指定主成分数量为 2
pca = PCA(n_components=2)# 使用 PCA 进行降维
X_pca = pca.fit_transform(X)# 打印降维后的数据形状
print(X_pca.shape)
13.2 LDA 降维

线性判别分析（LDA）是一种监督学习的降维方法，通常用于分类任务。

python">from sklearn.discriminant_analysis import LinearDiscriminantAnalysis as LDA# 创建 LDA 模型
lda = LDA(n_components=2)# 使用 LDA 进行降维
X_lda = lda.fit_transform(X, y)# 打印降维后的数据形状
print(X_lda.shape)

14. 模型持久化

在完成模型训练后，你可以使用 joblib 或 pickle 将模型保存为文件，之后可以加载该模型进行预测，而无需重新训练。

14.1 保存模型
python">import joblib# 保存模型到文件
joblib.dump(knn, 'knn_model.pkl')
14.2 加载模型
python"># 从文件中加载模型
loaded_model = joblib.load('knn_model.pkl')# 使用加载的模型进行预测
y_pred = loaded_model.predict(X_test)
print(f'Accuracy: {accuracy_score(y_test, y_pred)}')

15. 自定义估计器

除了使用 scikit-learn 提供的标准模型外，你还可以通过继承 BaseEstimator 和 ClassifierMixin 自定义自己的估计器。

python">from sklearn.base import BaseEstimator, ClassifierMixin
import numpy as npclass CustomClassifier(BaseEstimator, ClassifierMixin):def __init__(self, threshold=0.5):self.threshold = thresholddef fit(self, X, y):self.mean_ = np.mean(X, axis=0)return selfdef predict(self, X):return (np.mean(X, axis=1) > self.threshold).astype(int)# 创建并使用自定义分类器
clf = CustomClassifier(threshold=0.6)
clf.fit(X_train, y_train)
y_pred = clf.predict(X_test)
print(f'Accuracy: {accuracy_score(y_test, y_pred)}')

16. 完整示例：分类任务

下面是一个完整的例子，展示了如何加载数据、进行预处理、构建管道、训练模型、进行网格搜索、评估模型性能，并将模型保存。

python">import numpy as np
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split, GridSearchCV
from sklearn.preprocessing import StandardScaler
from sklearn.neighbors import KNeighborsClassifier
from sklearn.metrics import accuracy_score, classification_report
from sklearn.pipeline import Pipeline
import joblib# 加载数据集
iris = load_iris()
X = iris.data
y = iris.target# 数据划分
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)# 构建管道
pipeline = Pipeline([('scaler', StandardScaler()),  # 标准化('knn', KNeighborsClassifier())  # KNN 分类器
])# 定义参数网格
param_grid = {'knn__n_neighbors': [3, 5, 7],'knn__weights': ['uniform', 'distance']
}# 网格搜索
grid_search = GridSearchCV(pipeline, param_grid, cv=5)
grid_search.fit(X_train, y_train)# 最佳模型
print(f'Best parameters: {grid_search.best_params_}')# 在测试集上进行预测
y_pred = grid_search.predict(X_test)# 评估模型
print(f'Accuracy: {accuracy_score(y_test, y_pred)}')
print(classification_report(y_test, y_pred))# 保存最佳模型
joblib.dump(grid_search.best_estimator_, 'best_knn_model.pkl')

17. 总结

通过本教程，你已经了解了 Scikit-learn 的主要功能和使用方法，包括数据预处理、模型训练与评估、超参数调优、管道、特征选择、聚类、降维等。Scikit-learn 提供了强大且易用的 API，适合从简单的机器学习任务到更复杂的工作流构建。


http://www.ppmy.cn/embedded/115268.html

相关文章

Linux 动静态库

Linux 动静态库

目录 一.静态库 1.理解静态库 a.什么是静态库？ b.创建静态库的理论？ 2.打包静态库 3.静态库的使用方法 a.头文件找不着 b.链接报错——库函数文件找不着 4.将静态库文件写到系统目录下 a.直接拷贝 b.建立软链接 二.动态库 1.什么是动态库&am…
阅读更多...
Redis的一些通用指令

Redis的一些通用指令

首先我们需要先连接客户端服务器，此时我们需要通过redis-cli和redis服务器进行交互，输入ping来确保通路的流畅 （一）get和set redis中最核心的两个命令就是get和set，get就是根据key来取出对应value，set就是把…
阅读更多...
设计模式之组合模式例题

设计模式之组合模式例题

答案：C A 知识点：组合模式的意图：将对象组合成树型结构以表示“整体-部分”的层次结构，使得用户对单个对象和组合对象的使用具有一致性
阅读更多...
react native（expo）多语言适配

react native（expo）多语言适配

项目基于 expo框架 开发。请先配置好 expo 开发环境 1.引入i18n-js npx expo install i18n-js 2.新建languages文件夹，其中包括英文、中文等语种目录。结构如下： *.json文件为语种翻译后的json键值对，用于UI中引用; { "appName&q…
阅读更多...
Parallels Desktop 20(Mac虚拟机) v20.0.0 for Mac 最新破解版(支持M系列)

Parallels Desktop 20(Mac虚拟机) v20.0.0 for Mac 最新破解版(支持M系列)

Parallels Desktop 20 for Mac 正式发布，完全支持 macOS Sequoia 和 Windows 11 24H2，并且在企业版中引入了全新的管理门户。 据介绍，新版本针对 Windows、macOS 和 Linux 虚拟机进行了大量更新，最大的亮点是全新推出的 Parallels…
阅读更多...
Cubieboard2（三） 系统构建 —— WSL Ubuntu 中挂载 U 盘(SDCard)

Cubieboard2（三） 系统构建 —— WSL Ubuntu 中挂载 U 盘(SDCard)

文章目录 1 WSL Ubuntu 中挂载 U 盘(SDCard)2 usbipd 搭建虚拟机与宿主机 USB 通信桥梁3 WSL 内核添加 USB 设备驱动3.1 编译 WSL Linux 内核3.2 挂载 USB(SDCard) 设备 附录：WSL 操作命令附录：git 仓库检出 1 WSL Ubuntu 中挂载 U 盘(SDCard) Linux 驱动…
阅读更多...
聚焦于 Web 性能指标 TTI

聚焦于 Web 性能指标 TTI

在优化网站性能的过程中，我们经常遇到一个“为指标而优化”的困境。指标并不能真正反映用户体验，而应该最真实地反映用户行为。 在本节中，我们将研究 TTI（Time to Interactive）。在深入探讨这个话题之前，我…
阅读更多...
KG Structure as Prompt：利用知识图谱构建Prompt，提高大模型对因果关系的理解

KG Structure as Prompt：利用知识图谱构建Prompt，提高大模型对因果关系的理解

KG Structure as Prompt：利用知识图谱构建Prompt，提高大模型对因果关系的理解 秒懂大纲提出背景解法拆解创意视角中文意译 论文：Knowledge Graph Structure as Prompt: Improving Small Language Models Capabilities for Knowledge-based Ca…
阅读更多...
最新文章

Copyright @ 2022~2023