在我们用 Skorch 完成训练后,有没有办法绘制训练和验证的准确性net.fit(X_train, y_train)
。我们可以看到train_loss
, valid_loss
,valid_acc
但是train_acc
呢?谢谢你。
问问题
25 次
1 回答
0
这也在skorch 问题跟踪器中得到了回答,但简而言之,您可以简单地添加更多的得分器来提高训练的准确性:
net = NeuralNetClassifier(
# ...
callbacks=[
EpochScoring(scoring='accuracy', name='train_acc', on_train=True),
],
)
如果您在 jupyter notebook 中工作,您可以简单地运行
import matplotlib.pyplot as plt
plt.plot(net.history[:, 'train_acc'])
于 2021-08-24T14:34:25.880 回答