Source code for sklearndf.classification._classification_v1_6

"""
Additional implementation of :mod:`sklearndf.classification` loaded
from sklearn 1.0 onwards
"""

import logging

from sklearn.semi_supervised import SelfTrainingClassifier

from pytools.api import AllTracker

from .wrapper import MetaClassifierWrapperDF

log = logging.getLogger(__name__)

__all__ = [
    "SelfTrainingClassifierDF",
]

__imported_estimators = {name for name in globals().keys() if name.endswith("DF")}

#
# Ensure all symbols introduced below are included in __all__
#

__tracker = AllTracker(globals())

#
# threshold classifiers
#




class SelfTrainingClassifierDF(
    MetaClassifierWrapperDF[SelfTrainingClassifier],
    native=SelfTrainingClassifier,
):
    """
    Stub for DF wrapper of class ``SelfTrainingClassifier``.
    """



#
# validate __all__
#

__tracker.validate()


#
# validate that __all__ comprises all symbols ending in "DF", and no others
#

__estimators = {
    sym
    for sym in dir()
    if sym.endswith("DF")
    and sym not in __imported_estimators
    and not sym.startswith("_")
}
if __estimators != set(__all__):
    raise RuntimeError(
        "__all__ does not contain exactly all DF estimators; expected value is:\n"
        f"{__estimators}"
    )