一种训练用户分类模型的方法及装置

    公开(公告)号:CN120067842A

    公开(公告)日:2025-05-30

    申请号:CN202510220527.3

    申请日:2025-02-26

    Abstract: 本说明书实施例提供了一种训练用户分类模型的方法,包括:获取基于预训练模型分别初始化的第一分类模型、第二分类模型。利用第一分类模型,对用户集的第一子集中各用户进行类簇预测,得到类簇伪标签;以及,利用第二分类模型,对各用户进行M个用户群体的分类预测,得到分类伪标签。根据各用户的类簇伪标签和分类伪标签,从第一子集中确定出未知类别的第一用户。利用第一/第二分类模型,分别对该第一用户的增强样本进行预测,得到第一/第二预测结果;以第一预测结果接近该第一用户的类簇伪标签为目标,对第一分类模型进行更新;以第二预测结果中对应于M个用户群体的预测概率之和最大化为目标,对第二分类模型进行更新。

Patent Agency Ranking