weka - 具有概率估计的支持向量机

Question

我有一个二元分类问题，我正在用 SVM 解决。这些类在训练数据中是不平衡的。我现在需要获得后验概率输出，而不仅仅是二进制分数。我尝试通过 Weka 的 SMO 和 LibSVM 使用 Platt 缩放。对于这两种实现，我得到的结果，就少数类的 f1-measure 而言，比我只生成二元结果时更差。

您是否知道一种将 SVM 二元结果转换为概率的方法，该概率符合下一条规则:“概率 > = 0.5 当且仅当决策值 >= 0”。

这意味着在使用二元分类或概率时，每个样本获得的标签是相同的。

Answer 1

可以设置 SVM，以便它们输出类成员概率。您应该查看工具包的文档以了解如何启用它。

例如sckit-learn

When the constructor option probability is set to True, class membership probability estimates (from the methods predict_proba and predict_log_proba) are enabled.