问题描述
我正在对 fasttext 库中的分类模型使用自动调优代码,我无法确定它提供给您的最终模型是仅在训练集上训练还是在训练和验证集上训练。
例如,当你运行这个命令时:
>> ./fasttext supervised -input cooking.train -output model_cooking -autotune-validation cooking.valid
它是否会为您提供仅在 cooking.train
上或同时在 cooking.train
和 cooking.valid
上训练的模型。
提前致谢
解决方法
模型只在训练集上训练。验证集仅用于衡量模型的泛化能力,即检查模型是否真正学习了您希望它学习的概念,而不仅仅是记住训练数据。