对train函数增加注释