为自己的分类器编写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}