"""surrogate-forest: Random Forest with surrogate splits for native missing data handling."""

from .tree import SurrogateDecisionTreeClassifier, SurrogateDecisionTreeRegressor
from .ensemble import SurrogateRandomForestClassifier, SurrogateRandomForestRegressor
from .importance import impurity_importance, surrogate_importance, permutation_importance

__version__ = "0.1.0"

__all__ = [
    "SurrogateDecisionTreeClassifier",
    "SurrogateDecisionTreeRegressor",
    "SurrogateRandomForestClassifier",
    "SurrogateRandomForestRegressor",
    "impurity_importance",
    "surrogate_importance",
    "permutation_importance",
]
