如何进行机器学习分类,能够同时输出两类预测?

2024-09-28 22:23:25 发布

您现在位置:Python中文网/ 问答频道 /正文

业余机器学习程序员。我想执行一个分类任务,其中可以同时发生两个类预测。你知道吗

例如,在花卉图像分类中。除了能够对玫瑰或兰花的图像进行分类外,我还希望能够对同时包含玫瑰和兰花的图像进行分类。我必须训练我的模特把“玫瑰+兰花”区分为一个独立的班级吗?你知道吗

下面是任务的示例图像。你知道吗

Rose and Orchid Classification Problem


Tags: 图像机器示例分类程序员区分班级业余
2条回答

这称为多标签分类问题。有许多方法可以解决这个问题。学习有关multi-label classification的文档。你知道吗

An example with sklearn

在scikit中学习所有具有prob_a功能的分类器都有您的规范。此函数返回将每个类分配给输入x的概率。因此,您可以根据您的问题使用SVC、logistic回归、naivebayes、random forest或sciket learn中的任何解释分类器(如果您正在sciket learn中寻找指定的分类器)。你知道吗

当您找到每个类的prob_a时,如果两个最可能类之间的差异很接近,您可以使用两个最可能类引入输入。你知道吗

相关问题 更多 >