如何在 Java 中使用 spark ml 执行多标签分类答案

【问题标题】：How to perform Multilabel classification using spark ml in Java如何在 Java 中使用 spark ml 执行多标签分类
【发布时间】：2019-12-09 09:27:57
【问题描述】：

我建立了一个模型，该模型根据主题（功能）中的关键字将电子邮件分类为 10 个不同的work_categories（标签）。但是，此任务可能包含应将电子邮件分类为多个标签的可能性。

例如：主题为“服务证书并传输”的电子邮件应分为两个标签，即Servicing Worktype（标签1）和Transfer Worktype（标签2）。在我当前的程序中，这仅分配给标签1。无论如何我可以在java中使用spark ml将电子邮件分类到两个标签吗？

如果您有任何可以帮助我的建议或文档，请告诉我。谢谢

【问题讨论】：

【解决方案1】：

LogisticRegressionWithLBFGS和LogisticRegressionWithSGD都支持多标签分类：

为多项/二元 Logistic 训练分类模型使用有限内存 BFGS 进行回归。标准特征缩放和 L2 默认使用正则化。

或

使用二元逻辑回归训练分类模型随机梯度下降。默认情况下使用 L2 正则化，可以通过 LogisticRegressionWithSGD.optimizer 进行更改。

建议使用LogisticRegressionWithLBFGS。

【讨论】：