单元测试机器学习模型

数据挖掘 机器学习 数据挖掘 软件开发
2022-02-15 06:50:48

我被要求对我的机器学习模型(不是制作模型的代码)进行单元测试。由于我们实际上并不知道模型做出了什么预测,如何进行单元测试来检查模型的预测?这是怎么做到的?

编辑1:

我拥有的机器学习模型是根据患者的表格数据进行训练的。让我们举一个癌症预测的例子(我不能透露实际的例子,但这个例子非常接近)。它将来自各种测试的多次读数作为输入,并输出患者患癌症的距离或风险有多大。

编辑2:

有什么办法,比如测试每组输入(或)对抗性输入(确定模型将失败的输入)或极端输入情况的值范围。这方面的最佳实践是什么?

1个回答

在上述情况下,您可以收集带注释的数据(模型在训练期间看不到)并验证模型所做的预测。

另一种方法是,如果您是领域专家或对数据有足够的了解,您可以调整输入值并使用您的预期输出测试预测输出。