Python-sklearn模块-逻辑回归

导入模块

import numpy as np
from sklearn.linear_model import LogisticRegression

准备X的数据

X = np.array([
[60],
[20],
[30],
[80],
[59],
[90]
])

准备y的数据

y = np.array([1, 0, 0, 1, 0, 1])

创建逻辑回归模型

df = LogisticRegression()

填充数据并训练

df.fit(X, y)

准备用于测试的数据

X_test = np.array([
[62],
[87],
[39],
[45]
])

判断测试数据

y_predict = df.predict(X_test)
print(y_predict)
完整代码

import numpy as np
from sklearn.linear_model import LogisticRegression
X = np.array([
    [60],
    [20],
    [30],
    [80],
    [59],
    [90]
])
y = np.array([1, 0, 0, 1, 0, 1])
df = LogisticRegression()
df.fit(X, y)
X_test = np.array([
    [62],
    [87],
    [39],
    [45]
])
y_predict = df.predict(X_test)
print(y_predict)