【发布时间】:2020-08-26 02:27:29
【问题描述】:
我正在对 NER 使用 spaCy CLI train 命令,其中train_path 设置为训练数据集(训练集),dev_path 设置为评估数据集(测试集)。控制台中的打印输出显示了 NER Precision、Recall 和 F 分数。
但是,我不清楚分数是如何计算的。它们是来自模型预测的训练集(train-scores)还是来自测试集(test-scores)的分数?
我想确定在哪个 epoch 之后停止训练以防止过度拟合。目前,在 60 个 epoch 之后,Loss 仍然略有下降,而 Precision、Recall 和 F-score 仍然略有增加。在我看来,模型可能会记住训练数据,并且 P、R 和 F 分数是在训练集上计算的,因此会不断改进。
据我所知,训练的一个很好的停止点应该是在测试分数再次开始下降之前,即使训练分数不断增加。所以我想随着时间(时期)比较它们。
我的问题是:
- 在训练 train-scores 或 test-scores 时,分数会显示在控制台中吗?
- 以及如何访问另一个?
- 如果是训练分数,使用的测试集 (
dev_path) 是什么?
【问题讨论】:
标签: spacy named-entity-recognition