【问题标题】:sklearn logistic regression loss value during training训练期间的sklearn逻辑回归损失值
【发布时间】:2021-08-01 22:43:59
【问题描述】:

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

Python sklearn show loss values during training 有一个适用于 SGDRegressor 的示例,但不适用于逻辑回归。

【问题讨论】:

    标签: python scikit-learn logistic-regression


    【解决方案1】:

    我认为您应该将缩放器的参数从 **kwargs 更改为 **params

    old_stdout = sys.stdout
    sys.stdout = mystdout = StringIO()
    clf = LogisticRegression(**params)
    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:"+str(params)+".png")
    plt.xlabel("Time in epochs")
    plt.ylabel("Loss")
    plt.close()
    

    【讨论】:

      猜你喜欢
      • 2017-11-10
      • 2018-05-01
      • 2015-03-18
      • 2022-12-15
      • 1970-01-01
      • 2015-01-07
      • 2012-01-22
      • 2016-09-10
      • 2019-12-26
      相关资源
      最近更新 更多