带有标签train的模型