解释混淆矩阵

我需要创建一个分类模型来预测某事件的类别 – 1、2或3。我已经尝试了两种模型:多类决策树和多类神经网络。以下是每个模型的准确率得分和混淆矩阵。

多类决策树:

总体准确率 0.634平均准确率 0.756微平均精确度 0.634宏平均精确度 0.585184微平均召回率 0.634宏平均召回率 0.548334

混淆矩阵:

  1. 40.3% 53.6% 6.1%
  2. 6.6% 76.6% 16.8%
  3. 0.6% 51.8% 47.6%

多类神经网络:总体准确率 0.5865平均准确率 0.724333微平均精确度 0.5865宏平均精确度 0.583795微平均召回率 0.5865宏平均召回率 0.460215

混淆矩阵:

  1. 34.8% 63.5% 1.7%

  2. 2.9% 89.3% 7.7%

  3. 0.1% 85.9% 13.9%

    我认为这意味着在Class2上,这两个模型表现良好,尤其是神经网络模型。在其他类别上,决策树模型表现得更好,但仍然低于50%。

根据这些指标,我应该如何改进结果?谢谢。


回答:

删除id特征,同时检查并删除你认为对预测无价值的任何特征(如id等其他特征)或具有唯一值的特征。还要检查是否存在类别不平衡(数据中每个类别的样本数量是否有适当的平衡?)。然后尝试应用模型并调整参数以获得更好的结果。你可以使用交叉验证来获得更好的结果。

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注