ferrolearn-python 0.2.2

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

"""Test that all ferrolearn estimators pass sklearn's check_estimator."""

import pytest
from sklearn.utils.estimator_checks import parametrize_with_checks

from ferrolearn import (
    DecisionTreeClassifier,
    ElasticNet,
    GaussianNB,
    KMeans,
    KNeighborsClassifier,
    Lasso,
    LinearRegression,
    LogisticRegression,
    PCA,
    RandomForestClassifier,
    Ridge,
    StandardScaler,
)

ALL_ESTIMATORS = [
    LinearRegression(),
    Ridge(),
    Lasso(),
    ElasticNet(),
    LogisticRegression(),
    DecisionTreeClassifier(),
    RandomForestClassifier(n_estimators=5, random_state=42),
    KNeighborsClassifier(n_neighbors=3),
    GaussianNB(),
    StandardScaler(),
    PCA(n_components=2),
    KMeans(n_clusters=3, random_state=42, n_init=2),
]


@parametrize_with_checks(ALL_ESTIMATORS)
def test_estimator(estimator, check):
    check(estimator)