训练期间sklearn逻辑回归损失值

Question

有没有办法在训练逻辑回归时获得每次迭代的损失值？

Python sklearn 显示训练期间的损失值有一个 SGDRegressor 的工作示例，但不适用于逻辑回归。

Answer 1

我认为你应该更改参数 verbose 或将其删除。当您删除它时，它对我有用，默认情况下“verbose=0”。

old_stdout = sys.stdout
sys.stdout = mystdout = StringIO()
clf = LogisticRegression()
clf.fit(X_tr, y_tr)
sys.stdout = old_stdout
loss_history = mystdout.getvalue()
loss_list = []
for line in loss_history.split('\n'):
    if(len(line.split("loss: ")) == 1):
        continue
    loss_list.append(float(line.split("loss: ")[-1]))
plt.figure()
plt.plot(np.arange(len(loss_list)), loss_list)
plt.savefig("warmstart_plots/pure_LogRes:"+".png")
plt.xlabel("Time in epochs")
plt.ylabel("Loss")
plt.close()