Machine Learning for OpenCV 4 - Second Edition

Writing your own scikit-learn-based classifier in Python

Alternatively, you can write your own classifier using the scikit-learn library.

You can do this by importing BaseEstimator and ClassifierMixin. The latter will provide a corresponding score method, which works for all classifiers:

1. Optionally, first, you can overwrite the score method to provide your own metric score method:

In [1]: import numpy as np...     from sklearn.base import BaseEstimator, ClassifierMixin

2. Then, you can define a class that inherits from both BaseEstimator and ClassifierMixin:

In [2]: class MyClassifier(BaseEstimator, ClassifierMixin):...         """An example classifier"""

3. You need to provide a constructor, fit and predict methods. The constructor defines all the parameters ...