模型的训练时间受到好几方面的影响,比如所选的模型架构,设置的遍历次数,还有数据量的大小等。
对于模型架构,fasttext < CNN <= Self-Attention<= 长文本分类融合模型。遍历次数一般与训练时间成正比,也就是说 10 次的时间会是 5 次的两倍。为了节省用户时间,同时防止用户设置过大的遍历次数,我们内部内置了 earlystop 功能。具体的,模型在连续三次遍历的过程中效果都没有超过之前的最好结果就会自动停止。也就说说,虽然您设置的遍历次数是 30 次,但是可能在15 次的时候就停止了。
对于数据量大小,一般与训练时间成正比,数据量越多,训练时间越长。如果您希望很快看到结果,可以先尝试用一小部分数据进行训练,把所有流程跑通后,再用全量数据进行训练。