赞
踩
学习机器学习,就像学习任何新技能一样,最好的方法之一就是通过实战来巩固理论知识。鸢尾花分类项目是一个经典的入门项目,它不仅简单易懂,还能帮助我们掌握机器学习的基本步骤和方法。
鸢尾花数据集(Iris Dataset)早在1936年就由英国生物学家 Ronald A. Fisher 引入。这个数据集包含了150个鸢尾花样本,每个样本有4个特征(花萼长度、花萼宽度、花瓣长度和花瓣宽度)和1个目标变量(鸢尾花的品种)。这个数据集常用于教学和练习,是机器学习领域的 “Hello World” 项目。
本项目旨在使用鸢尾花数据集,通过数据准备、数据预处理、数据分割、模型选择、模型训练、模型评估、模型优化和模型部署等步骤,全面展示机器学习的基本流程。我们将使用 Python 和
scikit-learn
库,通过实际代码演示如何完成一个完整的机器学习项目。希望这篇文章能够帮助初学者快速上手机器学习,理解和掌握关键的技术和方法。
鸢尾花分类项目是机器学习领域中一个经典的入门项目,旨在通过分类算法将不同品种的鸢尾花进行分类。这个项目常用于教学和练习机器学习的基本概念。
鸢尾花数据集(Iris Dataset)由英国生物学家和统计学家 Ronald A. Fisher 于 1936 年在他的论文《The use of multiple measurements in taxonomic problems》中首次引入。这个数据集包含了 150 个鸢尾花样本,每个样本有 4 个特征和 1 个目标变量。
使用机器学习算法,根据鸢尾花的四个特征来预测其所属的品种。
以下是机器学习大致的基本步骤,我们后面实战部分也将是参考此步骤来进行的:
鸢尾花数据集作为机器学习领域中最著名的数据集之一,常用于入门级教学和算法验证。
我们可以通过以下地址对该数据集进行下载:https://archive.ics.uci.edu/dataset/53/iris。下载后,导入:
import pandas as pd
# 指定列名
feature_names = [
"sepal length (cm)",
"sepal width (cm)",
"petal length (cm)",
"petal width (cm)"
]
# 加载数据
data = pd.read_csv('path/to/iris.data', header=None, names=feature_names + ['class'])
# 查看数据
print(data.head())
或者,我们也可以使用 scikit-learn
库自带的标准数据集进行导入 from sklearn.datasets import load_iris
import pandas as pd
from sklearn.datasets import load_iris
# 加载数据集
iris = load_iris() # 使用 load_iris 函数加载鸢尾花数据集,返回一个 Bunch 对象
data = pd.DataFrame(data=iris.data, columns=iris.feature_names) # 将数据集的特征数据转换为 pandas DataFrame,并指定列名为特征名称
data['target'] = iris.target # 在 DataFrame 中添加一列 'target',对应每个样本的目标标签
data['target_names'] = data['target'].apply(lambda x: iris.target_names[x]) # 添加一列 'target_names',通过映射 'target' 列的值来获取对应的目标名称
# 查看数据
print(data.head()) # 查看数据集前五行数据
print('-')
print(data.info()) # 查看数据集的基本信息
print('-')
print(data.describe()) # 查看数据集的统计信息
打印结果:
sepal length (cm) sepal width (cm) ... target target_names
0 5.1 3.5 ... 0 setosa
1 4.9 3.0 ... 0 setosa
2 4.7 3.2 ... 0 setosa
3 4.6 3.1 ... 0 setosa
4 5.0 3.6 ... 0 setosa
[5 rows x 6 columns]
-
<class 'pandas.core.frame.DataFrame'>
RangeIndex: 150 entries, 0 to 149
Data columns (total 6 columns):
# Column Non-Null Count Dtype
--- ------ -------------- -----
0 sepal length (cm) 150 non-null float64
1 sepal width (cm) 150 non-null float64
2 petal length (cm) 150 non-null float64
3 petal width (cm) 150 non-null float64
4 target 150 non-null int64
5 target_names 150 non-null object
dtypes: float64(4), int64(1), object(1)
memory usage: 7.2+ KB
None
-
sepal length (cm) sepal width (cm) ... petal width (cm) target
count 150.000000 150.000000 ... 150.000000 150.000000
mean 5.843333 3.057333 ... 1.199333 1.000000
std 0.828066 0.435866 ... 0.762238 0.819232
min 4.300000 2.000000 ... 0.100000 0.000000
25% 5.100000 2.800000 ... 0.300000 0.000000
50% 5.800000 3.000000 ... 1.300000 1.000000
75% 6.400000 3.300000 ... 1.800000 2.000000
max 7.900000 4.400000 ... 2.500000 2.000000
[8 rows x 5 columns]
通过数据可视化手段,探索数据的分布和特征之间的关系。使用 seaborn
的 pairplot
函数绘制特征之间的分布图,以便于观察数据的分布情况和不同类别之间的关系。
import seaborn as sns
import matplotlib.pyplot as plt
# 绘制特征分布图
sns.pairplot(data, hue='target', markers=["o", "s", "D"]) # 使用 seaborn 库的 pairplot 函数绘制散点图矩阵,根据 'target' 列着色,并使用不同的标记
plt.show() # 显示图形
打印结果:
对数据进行清洗和转换,如处理缺失值、标准化或归一化特征等。
# 检查是否有缺失值
print(data.isnull().sum()) # 打印每列的缺失值数量,以检查数据集中是否存在缺失值
# 标准化特征值
scaler = StandardScaler() # 创建一个 StandardScaler 对象
data[iris.feature_names] = scaler.fit_transform(data[iris.feature_names]) # 对特征列进行标准化处理,使其均值为 0,标准差为 1
# 查看标准化后的数据
print(data.head()) # 打印标准化后的数据集的前五行,查看标准化效果
打印结果:
sepal length (cm) sepal width (cm) ... petal width (cm) target
count 150.000000 150.000000 ... 150.000000 150.000000
mean 5.843333 3.057333 ... 1.199333 1.000000
std 0.828066 0.435866 ... 0.762238 0.819232
min 4.300000 2.000000 ... 0.100000 0.000000
25% 5.100000 2.800000 ... 0.300000 0.000000
50% 5.800000 3.000000 ... 1.300000 1.000000
75% 6.400000 3.300000 ... 1.800000 2.000000
max 7.900000 4.400000 ... 2.500000 2.000000
[8 rows x 5 columns]
-
sepal length (cm) 0
sepal width (cm) 0
petal length (cm) 0
petal width (cm) 0
target 0
target_names 0
dtype: int64
sepal length (cm) sepal width (cm) ... target target_names
0 -0.900681 1.019004 ... 0 setosa
1 -1.143017 -0.131979 ... 0 setosa
2 -1.385353 0.328414 ... 0 setosa
3 -1.506521 0.098217 ... 0 setosa
4 -1.021849 1.249201 ... 0 setosa
[5 rows x 6 columns]
将数据分成训练集和测试集。训练集用于训练模型,测试集用于评估模型的性能。
分割数据集的主要目的是为了评估模型的性能。通过将数据集划分为训练集和测试集,我们可以在训练集上训练模型,并在测试集上评估模型的泛化能力。这有助于检测模型是否过拟合或欠拟合,并提供对模型性能的更真实的评估。
from sklearn.model_selection import train_test_split
# 分割数据集
X = data[iris.feature_names] # 特征数据,包含所有特征列
y = data['target'] # 目标数据,包含目标标签
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3,
random_state=42) # 将数据集按 70% 训练集和 30% 测试集进行分割,设置随机种子为 42 以确保可重复性
# 查看分割后的数据集大小
print(f"训练集大小: {X_train.shape[0]}, 测试集大小: {X_test.shape[0]}") # 打印训练集和测试集的大小
打印结果:
训练集大小: 105, 测试集大小: 45
选择合适的机器学习算法来训练模型,如 K近邻(KNN)、决策树、逻辑回归、支持向量机(SVM)等。
这里选择 K 近邻(KNN)算法。
from sklearn.neighbors import KNeighborsClassifier
# 初始化 KNN 分类器
knn = KNeighborsClassifier(n_neighbors=3) # 创建一个 K 近邻分类器对象,设置近邻数为 3
使用训练数据来训练模型。
# 训练模型
knn.fit(X_train, y_train) # 使用训练数据训练 K 近邻分类器模型
使用测试数据评估模型的性能,查看模型的准确率、混淆矩阵等评估指标。
评估指标解释:
pfrom sklearn.metrics import accuracy_score, classification_report, confusion_matrix
# 预测测试集
y_pred = knn.predict(X_test) # 使用训练好的模型对测试数据进行预测,得到预测标签
# 评估模型
print(f"Accuracy: {accuracy_score(y_test, y_pred)}") # 计算并打印模型在测试集上的准确率
print('-')
print("Classification Report:\n", classification_report(y_test, y_pred)) # 打印分类报告,包括精确率、召回率、F1 分数等
print('-')
print("Confusion Matrix:\n", confusion_matrix(y_test, y_pred)) # 打印混淆矩阵,显示真实标签和预测标签之间的关系
print('-')
打印结果:
Accuracy: 1.0
-
Classification Report:
precision recall f1-score support
0 1.00 1.00 1.00 19
1 1.00 1.00 1.00 13
2 1.00 1.00 1.00 13
accuracy 1.00 45
macro avg 1.00 1.00 1.00 45
weighted avg 1.00 1.00 1.00 45
-
Confusion Matrix:
[[19 0 0]
[ 0 13 0]
[ 0 0 13]]
根据评估结果调整模型参数,优化模型性能。例如,通过网格搜索优化 KNN 模型的参数。
PS:网格搜索是一种系统化的方法,用于调整模型的超参数以提高模型的性能。通过交叉验证和不同参数组合的评估,网格搜索可以找到最佳的参数配置,从而提高模型的预测准确性和泛化能力。
from sklearn.model_selection import GridSearchCV
# 定义参数范围
param_grid = {'n_neighbors': range(1, 20)} # 创建一个参数字典,设置 'n_neighbors' 参数的取值范围为 1 到 19
# 网格搜索
grid_search = GridSearchCV(KNeighborsClassifier(), param_grid, cv=5) # 创建一个 GridSearchCV 对象,使用 KNeighborsClassifier 和定义的参数范围,设置交叉验证折数为 5
grid_search.fit(X_train, y_train) # 使用训练数据进行网格搜索,以找到最佳参数组合
# 最优参数
print(f"Best parameters: {grid_search.best_params_}") # 打印通过网格搜索找到的最优参数
print('-')
# 使用最优参数训练模型
knn_best = grid_search.best_estimator_ # 获取使用最优参数训练的最佳模型
y_pred_best = knn_best.predict(X_test) # 使用最佳模型对测试数据进行预测
# 评估模型
print(f"Accuracy(best): {accuracy_score(y_test, y_pred)}") # (优化后)计算并打印模型在测试集上的准确率
print('-')
print("Classification Report(best):\n", classification_report(y_test, y_pred)) # (优化后)打印分类报告,包括精确率、召回率、F1 分数等
print('-')
print("Confusion Matrix(best):\n", confusion_matrix(y_test, y_pred)) # (优化后)打印混淆矩阵,显示真实标签和预测标签之间的关系
print('-')
打印结果:
Best parameters: {'n_neighbors': 1}
-
Accuracy(best): 1.0
-
Classification Report(best):
precision recall f1-score support
0 1.00 1.00 1.00 19
1 1.00 1.00 1.00 13
2 1.00 1.00 1.00 13
accuracy 1.00 45
macro avg 1.00 1.00 1.00 45
weighted avg 1.00 1.00 1.00 45
-
Confusion Matrix(best):
[[19 0 0]
[ 0 13 0]
[ 0 0 13]]
将训练好的模型应用到实际问题中进行预测。例如,预测新的鸢尾花样本的类别。
PS:为什么要标准化新样本?标准化新样本是必要的步骤,因为模型在训练时使用的是标准化后的数据。如果新样本没有经过相同的标准化处理,特征值的尺度可能会与训练数据不同,从而导致预测不准确。通过使用相同的 StandardScaler
对新样本进行标准化,可以确保特征值在相同的尺度上,保证模型的预测效果。
# 新样本数据
new_samples = [[5.1, 3.5, 1.4, 0.2], [6.7, 3.0, 5.2, 2.3]]
# 将新样本数据转换为 DataFrame,并指定特征名称
new_samples_df = pd.DataFrame(new_samples, columns=iris.feature_names)
# 标准化新样本
new_samples_scaled = scaler.transform(new_samples_df) # 使用之前的标准化器 scaler 对新样本进行标准化处理,使其与训练数据具有相同的尺度
# 预测新样本的类别
predictions = knn_best.predict(new_samples_scaled) # 使用最佳模型 knn_best 对标准化后的新样本进行类别预测
print(f"新样本的预测类别: {predictions}") # 打印新样本的预测类别
打印结果:
新样本的预测类别: [0 2]
打印结果显示新样本的预测类别分别为 [0, 2]
,即第一个新样本被预测为类别 0,第二个新样本被预测为类别 2。结合鸢尾花数据集的类别名称,这意味着:
Iris-setosa
Iris-virginica
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
from sklearn.datasets import load_iris
from sklearn.preprocessing import StandardScaler
from sklearn.model_selection import train_test_split, GridSearchCV
from sklearn.neighbors import KNeighborsClassifier
from sklearn.metrics import accuracy_score, classification_report, confusion_matrix
# 加载数据集
iris = load_iris() # 使用 load_iris 函数加载鸢尾花数据集,返回一个 Bunch 对象
data = pd.DataFrame(data=iris.data, columns=iris.feature_names) # 将数据集的特征数据转换为 pandas DataFrame,并指定列名为特征名称
data['target'] = iris.target # 在 DataFrame 中添加一列 'target',对应每个样本的目标标签
data['target_names'] = data['target'].apply(
lambda x: iris.target_names[x]) # 添加一列 'target_names',通过映射 'target' 列的值来获取对应的目标名称
# 查看数据
print(data.head()) # 查看数据集前五行数据
print('-')
print(data.info()) # 查看数据集的基本信息
print('-')
print(data.describe()) # 查看数据集的统计信息
print('-')
# 绘制特征分布图
sns.pairplot(data, hue='target', markers=["o", "s", "D"]) # 使用 seaborn 库的 pairplot 函数绘制散点图矩阵,根据 'target' 列着色,并使用不同的标记
plt.show() # 显示图形
# 检查是否有缺失值
print(data.isnull().sum()) # 打印每列的缺失值数量,以检查数据集中是否存在缺失值
# 标准化特征值
scaler = StandardScaler() # 创建一个 StandardScaler 对象
data[iris.feature_names] = scaler.fit_transform(data[iris.feature_names]) # 对特征列进行标准化处理,使其均值为 0,标准差为 1
# 查看标准化后的数据
print(data.head()) # 打印标准化后的数据集的前五行,查看标准化效果
print('-')
# 分割数据集
X = data[iris.feature_names] # 特征数据,包含所有特征列
y = data['target'] # 目标数据,包含目标标签
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3,
random_state=42) # 将数据集按 70% 训练集和 30% 测试集进行分割,设置随机种子为 42 以确保可重复性
# 查看分割后的数据集大小
print(f"训练集大小: {X_train.shape[0]}, 测试集大小: {X_test.shape[0]}") # 打印训练集和测试集的大小
print('-')
# 初始化 KNN 分类器
knn = KNeighborsClassifier(n_neighbors=3) # 创建一个 K 近邻分类器对象,设置近邻数为 3
# 训练模型
knn.fit(X_train, y_train) # 使用训练数据训练 K 近邻分类器模型
# 预测测试集
y_pred = knn.predict(X_test) # 使用训练好的模型对测试数据进行预测,得到预测标签
# 评估模型
print(f"Accuracy: {accuracy_score(y_test, y_pred)}") # 计算并打印模型在测试集上的准确率
print('-')
print("Classification Report:\n", classification_report(y_test, y_pred)) # 打印分类报告,包括精确率、召回率、F1 分数等
print('-')
print("Confusion Matrix:\n", confusion_matrix(y_test, y_pred)) # 打印混淆矩阵,显示真实标签和预测标签之间的关系
print('-')
# 定义参数范围
param_grid = {'n_neighbors': range(1, 20)} # 创建一个参数字典,设置 'n_neighbors' 参数的取值范围为 1 到 19
# 网格搜索
grid_search = GridSearchCV(KNeighborsClassifier(), param_grid, cv=5) # 创建一个 GridSearchCV 对象,使用 KNeighborsClassifier 和定义的参数范围,设置交叉验证折数为 5
grid_search.fit(X_train, y_train) # 使用训练数据进行网格搜索,以找到最佳参数组合
# 最优参数
print(f"Best parameters: {grid_search.best_params_}") # 打印通过网格搜索找到的最优参数
print('-')
# 使用最优参数训练模型
knn_best = grid_search.best_estimator_ # 获取使用最优参数训练的最佳模型
y_pred_best = knn_best.predict(X_test) # 使用最佳模型对测试数据进行预测
# 评估模型
print(f"Accuracy(best): {accuracy_score(y_test, y_pred)}") # (优化后)计算并打印模型在测试集上的准确率
print('-')
print("Classification Report(best):\n", classification_report(y_test, y_pred)) # (优化后)打印分类报告,包括精确率、召回率、F1 分数等
print('-')
print("Confusion Matrix(best):\n", confusion_matrix(y_test, y_pred)) # (优化后)打印混淆矩阵,显示真实标签和预测标签之间的关系
print('-')
# 新样本数据
new_samples = [[5.1, 3.5, 1.4, 0.2], [6.7, 3.0, 5.2, 2.3]] # 定义新的样本数据,每个样本包含四个特征值
# 将新样本数据转换为 DataFrame,并指定特征名称
new_samples_df = pd.DataFrame(new_samples, columns=iris.feature_names)
# 标准化新样本
new_samples_scaled = scaler.transform(new_samples_df) # 使用之前的标准化器 scaler 对新样本进行标准化处理,使其与训练数据具有相同的尺度
# 预测新样本的类别
predictions = knn_best.predict(new_samples_scaled) # 使用最佳模型 knn_best 对标准化后的新样本进行类别预测
print(f"新样本的预测类别: {predictions}") # 打印新样本的预测类别
通过这段代码,我们展示了如何加载鸢尾花数据集、进行标准化处理、分割数据集、使用网格搜索优化 K 近邻分类器、训练模型、评估模型以及对新样本进行预测。最终的预测结果 [0, 2]
表明第一个新样本被预测为 Iris-setosa
,第二个新样本被预测为 Iris-virginica
,这验证了整个机器学习过程的正确性和有效性。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。