添加训练模型数据读取类