from sklearn.linear_model import LogisticRegression as skllogitic
from sklearn.datasets import make_classification
from cuml.linear_model import LogisticRegression


def test_logistic_reg():
    X, y = make_classification(
        random_state=3, n_classes=5, n_features=8, n_informative=5
    )
    reg_skl = skllogitic()
    reg_skl.fit(X, y)
    d_skl = reg_skl.decision_function(X)

    reg = LogisticRegression()
    reg.fit(X, y)
    d_cu = reg.decision_function(X)

    print(d_skl.shape, d_cu.shape)
    # (100, 5), (5, 100)

test_logistic_reg()