机器学习模型泛化

Question

我是机器学习的新手，我想就模型泛化提出一个问题。就我而言，我将生产一些机械零件，我对控制输入参数以获得最终零件的某些属性感兴趣。

更具体地说，我对 8 个参数（比如 P1、P2、...、P8）感兴趣。为了优化生产所需零件的数量以最大化探索的参数组合，我将问题分为两组。对于第一组片段，我将改变前 4 个参数（P1 ... P4），而其他参数保持不变。在第二种情况下，我会做相反的事情（变量 P5 ... P8 和常量 P1 ... P4）。

所以我想知道是否可以制作一个具有八个参数作为输入的单一模型来预测最终部件的属性。我问是因为我没有同时改变所有 8 个变量，我想也许我必须为每组参数做 1 个模型，并且 2 个不同模型的预测不能相互关联.

提前致谢。

Answer 1

在大多数情况下，两个不同的模型比一个大模型具有更好的准确性。原因是在局部模型中，模型只会查看 4 个特征，并且能够识别其中的模式进行预测。

但这种特殊方法肯定无法扩展。现在你只有两组数据，但如果它增加并且你有 20 组数据怎么办。您无法在生产中创建和维护 20 个 ML 模型。

最适合您的情况的方法需要进行一些试验。从数据中随机抽取样本并训练 ML 模型。取一个大模型和两个局部模型并评估它们的性能。不仅是准确性，还有他们的 F1 分数、AUC-PR 和 ROC 曲线，以找出最适合您的方法。如果您没有看到性能大幅下降，那么整个数据集的一个大模型将是一个更好的选择。如果您知道您的数据将始终分为这两组并且您不关心可伸缩性，那么请使用两个本地模型。