weka - 带有概率估计的 SVM

Question

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

您是否知道将 SVM 二进制结果转换为保持下一条规则的概率的方法：“当且仅当决策值 >= 0 时，概率 > = 0.5”。

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

Answer 1

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

例如sckit-learn

当构造函数选项概率设置为 True 时，将启用类成员概率估计（来自方法 predict_proba 和 predict_log_proba）。