科技 > 人工智能 > 机器学习

随机森林、逻辑回归、多层感知机和极端梯度提升(XGBoost)是四种常见的机器学习模型-代码模型评价可视化

120人参与 2024-08-06 机器学习

**随机森林、逻辑回归、多层感知机和极端梯度提升(xgboost)**是四种常见的机器学习模型,每种模型都有其独特的算法原理和应用场景。下面将分别对这些模型进行讲解,并提供相应的python代码示例。

1,随机森林
随机森林是一种集成学习方法,它通过构建多个决策树并进行投票或平均来改善模型的预测能力。随机森林引入了随机性,不仅在选择样本时进行抽样(装袋法),还在选择分裂特征时引入随机选择。这种方法可以提高模型的泛化能力,减少过拟合

代码示例:
from sklearn.ensemble import randomforestclassifier
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split

#加载数据
iris = load_iris()
x, y = iris.data, iris.target

#划分数据集
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.3, random_state=42)

#初始化随机森林模型
rf = randomforestclassifier(n_estimators=100, random_state=42)

#训练模型
rf.fit(x_train, y_train)

#进行预测
predictions = rf.predict(x_test)

2.逻辑回归
逻辑回归是一种广泛用于分类任务的线性模型,尤其是二分类问题。逻辑回归通过sigmoid函数将线性回归的输出映射到(0,1)区间,可以解释为概率输出。逻辑回归简单、高效,适用于线性可分的数据集。

逻辑回归代码示例:
from sklearn.linear_model import logisticregression
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split

#加载数据
iris = load_iris()
x, y = iris.data, iris.target

#仅使用两个类别进行二分类
x, y = x[y != 2], y[y != 2]

#划分数据集
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.3, random_state=42)

#初始化逻辑回归模型
lr = logisticregression()

#训练模型
lr.fit(x_train, y_train)

#进行预测
predictions = lr.predict(x_test)

3. 多层感知机(multilayer perceptron, mlp)
多层感知机是一种前馈人工神经网络,包含输入层、一个或多个隐藏层和输出层。每个层都由神经元组成,神经元之间通过权重连接。mlp通过反向传播算法进行训练,可以捕获数据中的非线性关系,适用于复杂的分类和回归问题。

多层感知机代码示例:
from sklearn.neural_network import mlpclassifier
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split

#加载数据
iris = load_iris()
x, y = iris.data, iris.target

#划分数据集
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.3, random_state=42)

#初始化多层感知机模型
mlp = mlpclassifier(hidden_layer_sizes=(100,), max_iter=500, random_state=42)

#训练模型
mlp.fit(x_train, y_train)

#进行预测
predictions = mlp.predict(x_test)

4. 极端梯度提升(xgboost)
极端梯度提升是一种高效的集成学习算法,通过逐步添加树模型并利用梯度提升方法优化模型来提高性能。xgboost对于处理缺失值、正则化以防止过拟合以及在树的构建过程中进行列抽样等方面进行了优化,广泛应用于各种机器学习竞赛和实际问题。

xgboost代码示例:
import xgboost as xgb
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split

#加载数据
iris = load_iris()
x, y = iris.data, iris.target

#划分数据集
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.3, random_state=42)

#初始化xgboost模型
xgb_model = xgb.xgbclassifier(use_label_encoder=false, eval_metric=‘mlogloss’)

#训练模型
xgb_model.fit(x_train, y_train)

#进行预测
predictions = xgb_model.predict(x_test)

补充模型评估与可视化:
1. 模型评估

评估模型通常涉及计算一些指标,如准确率、精确率、召回率和f1分数。对于分类问题,混淆矩阵也是一个非常有用的工具,因为它可以显示模型在各个类别上的性能。

评估代码示例:

from sklearn.metrics import accuracy_score, classification_report, confusion_matrix
import seaborn as sns
import matplotlib.pyplot as plt

#假设predictions是模型预测结果,y_test是真实标签

#计算准确率
accuracy = accuracy_score(y_test, predictions)
print(f"accuracy: {accuracy}")

#显示分类报告
print(classification_report(y_test, predictions))

生成混淆矩阵
conf_matrix = confusion_matrix(y_test, predictions)

#可视化混淆矩阵
sns.heatmap(conf_matrix, annot=true, fmt=‘d’, cmap=‘blues’)
plt.xlabel(‘predicted label’)
plt.ylabel(‘true label’)
plt.title(‘confusion matrix’)
plt.show()

2. 可视化
可视化可以帮助我们更直观地理解模型的工作原理和性能。对于决策树或随机森林,我们可以可视化树结构;对于模型的结果,可以使用混淆矩阵、roc曲线等进行可视化。

决策树可视化(适用于随机森林中的单个树):

from sklearn.tree import plot_tree

#假设rf是训练好的随机森林模型
#选择随机森林中的第一棵树
estimator = rf.estimators_[0]

plt.figure(figsize=(20,10))
plot_tree(estimator, filled=true, feature_names=iris.feature_names, class_names=iris.target_names, rounded=true)
plt.show()

(0)
打赏 微信扫一扫 微信扫一扫

您想发表意见!!点此发布评论

推荐阅读

机器学习——逻辑回归

08-06

智能风控体系之逻辑回归

08-06

逻辑回归

08-06

机器学习个人作业:回归分析

08-06

机器学习:逻辑回归

08-06

最大似然估计和逻辑回归

08-06

猜你喜欢

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论