一种基于BERT的体能训练数据特征提取方法

摘要

本发明实施例提供一种基于BERT的体能训练数据特征提取方法，将进行体能训练的人群日常训练数据的80％作为模型的训练集，10％作为验证集，10％作为测试集，进行K折交叉验证，在BERT部分，使用word2vec预训练体能训练项目的embedding，将训练进行聚类；改进MLM，将原始直接mask包括属性在内的整个训练信息的形式，改为仅进行mask训练的一部分信息；融合部分将满足预设条件的开展体能训练人群的年龄、性别、生理状态和心理状态分布融入到网络当中；输出部分，对受训人员的年龄、性别、生理状态、心理状态分布进行分类预测，输出采用不同训练实现有效训练的用户类别概率；将模型训练至收敛后结束。