从现有系数创建sklearn.linear_model.LogisticRegression实例

时间:2014-06-26 19:36:30

标签: python scikit-learn logistic-regression

是否可以根据现有系数创建这样的实例,这些系数是在不同的实现(例如Java)中计算的?

我尝试创建一个实例,然后直接设置coef_和intercept_,它似乎有效,但我不确定这里是否存在缺陷,或者我是否会破坏某些内容。

1 个答案:

答案 0 :(得分:1)

是的,它运作正常:

import numpy as np
from scipy.stats import norm
from sklearn.linear_model import LogisticRegression
import json
x = np.arange(10)[:, np.newaxis]
y = np.array([0,0,0,1,0,0,1,1,1,1])
# training one logistic regression
model1 = LogisticRegression(C=10, penalty='l1').fit(x, y)
# serialize coefficients (imitate loading from storage)
encoded = json.dumps((model1.coef_.tolist(), model1.intercept_.tolist(), model1.penalty, model1.C))
print(encoded)
decoded = json.loads(encoded)
# using coefficients in another regression
model2 = LogisticRegression()
model2.coef_ = np.array(decoded[0])
model2.intercept_ = np.array(decoded[1])
model2.penalty = decoded[2]
model2.C = decoded[3]
# resulting predictions are identical
print(model1.predict_proba(x) == model2.predict_proba(x))

输出:

[[[0.7558780101653273]], [-3.322083150375962], "l1", 10]
[[ True  True]
 [ True  True]
 [ True  True]
 [ True  True]
 [ True  True]
 [ True  True]
 [ True  True]
 [ True  True]
 [ True  True]
 [ True  True]]

因此对原始模型和重新创建的模型的预测确实相同。