为自己的分类器编写 scikit-learn 测试

Question

我正在编写自己的分类器，用于使用 scikit-learn 进行新奇检测。 现在，为了能够在框架内无缝使用它，我需要它通过 check_estimator() 测试。

我的问题是我的分类器只返回两个标签（0 或 1，取决于它是否认为输入对应于异常值）。

但是随后 utils/estimator_checks.py 中的测试 check_classifiers_classes() 失败，因为它期望分类器返回两个以上的类。 在 scikit-learn 中实现/测试新奇检测器的正确方法是什么？

Answer 1

您应该在分类器中添加一个方法来告诉 sklearn 这是一个二元分类器

def _more_tags(self) -> dict:
    return {"binary_only": True}