一种模型训练方法、磁盘预测方法、装置及电子设备

基本信息

申请号 CN202011334753.8 申请日 -
公开(公告)号 CN112395179A 公开(公告)日 2021-02-23
申请公布号 CN112395179A 申请公布日 2021-02-23
分类号 G06F11/34(2006.01)I;G06F11/30(2006.01)I 分类 计算;推算;计数;
发明人 黄泽;王梦秋;胡太祥;张泽覃 申请(专利权)人 创新奇智(西安)科技有限公司
代理机构 北京超凡宏宇专利代理事务所(特殊普通合伙) 代理人 钟扬飞
地址 710000陕西省西安市高新区鱼化街办软件新城天谷八路528号国家电子商务示范基地东区530
法律状态 -

摘要

摘要 本申请涉及一种模型训练方法、磁盘预测方法、装置及电子设备,属于计算机领域。方法包括获取表征磁盘故障的负样本数据以及表征磁盘正常的正样本数据;基于负样本数据以及正样本数据对初始模型进行N次迭代训练,得到能预测磁盘是否故障的预测模型;在第i次迭代训练过程中,以第i‑1次迭代训练所得的模型在正样本数据上产生的损失值对正样本进行下采样,并将采样得到的正样本数据以及负样本数据对第i‑1次迭代训练所得的模型进行训练。该方法在对正样本进行下采样时,由于每次迭代训练所得的模型在正样本数据上产生的损失值均不同,因此每次采样的正样本数据也不同,使得在采样时可以基于分类难易度,更多关注困难的样本,从而提高了模型的精度。