ferrolearn-python 0.4.0

Python bindings for ferrolearn via PyO3
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125

"""ferrolearn — scikit-learn compatible Rust ML models."""

# Original 12 estimators
from ferrolearn._regressors import ElasticNet, Lasso, LinearRegression, Ridge
from ferrolearn._classifiers import (
    DecisionTreeClassifier,
    GaussianNB,
    KNeighborsClassifier,
    LogisticRegression,
    RandomForestClassifier,
)
from ferrolearn._transformers import PCA, StandardScaler
from ferrolearn._clusterers import KMeans

# Phase-2 expansion (~40 more)
from ferrolearn._extras import (
    AdaBoostClassifier,
    AgglomerativeClustering,
    ARDRegression,
    BaggingClassifier,
    BayesianRidge,
    BernoulliNB,
    Birch,
    ComplementNB,
    DBSCAN,
    DecisionTreeRegressor,
    ExtraTreeClassifier,
    ExtraTreesClassifier,
    ExtraTreesRegressor,
    FactorAnalysis,
    FastICA,
    GaussianMixture,
    GradientBoostingClassifier,
    GradientBoostingRegressor,
    HistGradientBoostingClassifier,
    HistGradientBoostingRegressor,
    HuberRegressor,
    IncrementalPCA,
    KernelPCA,
    KernelRidge,
    KNeighborsRegressor,
    LinearSVC,
    MaxAbsScaler,
    MinMaxScaler,
    MiniBatchKMeans,
    MultinomialNB,
    NMF,
    NearestCentroid,
    Nystroem,
    PowerTransformer,
    QuadraticDiscriminantAnalysis,
    QuantileRegressor,
    RBFSampler,
    RandomForestRegressor,
    RidgeClassifier,
    RobustScaler,
    SparsePCA,
    TruncatedSVD,
)

__all__ = [
    # Original 12
    "LinearRegression",
    "Ridge",
    "Lasso",
    "ElasticNet",
    "LogisticRegression",
    "DecisionTreeClassifier",
    "RandomForestClassifier",
    "KNeighborsClassifier",
    "GaussianNB",
    "StandardScaler",
    "PCA",
    "KMeans",
    # Linear extras
    "BayesianRidge",
    "ARDRegression",
    "HuberRegressor",
    "QuantileRegressor",
    "RidgeClassifier",
    "LinearSVC",
    "QuadraticDiscriminantAnalysis",
    # Bayes extras
    "MultinomialNB",
    "BernoulliNB",
    "ComplementNB",
    # Tree extras
    "DecisionTreeRegressor",
    "RandomForestRegressor",
    "ExtraTreeClassifier",
    "ExtraTreesClassifier",
    "ExtraTreesRegressor",
    "AdaBoostClassifier",
    "GradientBoostingClassifier",
    "GradientBoostingRegressor",
    "HistGradientBoostingClassifier",
    "HistGradientBoostingRegressor",
    "BaggingClassifier",
    # Neighbors extras
    "KNeighborsRegressor",
    "NearestCentroid",
    # Cluster extras
    "MiniBatchKMeans",
    "DBSCAN",
    "AgglomerativeClustering",
    "Birch",
    "GaussianMixture",
    # Decomp extras
    "IncrementalPCA",
    "TruncatedSVD",
    "FastICA",
    "NMF",
    "KernelPCA",
    "SparsePCA",
    "FactorAnalysis",
    # Preprocess extras
    "MinMaxScaler",
    "MaxAbsScaler",
    "RobustScaler",
    "PowerTransformer",
    # Kernel extras
    "KernelRidge",
    "Nystroem",
    "RBFSampler",
]