用于多类分类的 pytorch 类权重

Posted 2023-03-27

【中文标题】用于多类分类的 pytorch 类权重

【英文标题】：pytorch class weights for multi class classification

【发布时间】：2022-01-06 21:41:05

【问题描述】：

我正在使用类权重进行多类分类，使用 sklearn 的 compute_weight 函数和 pytorch 来训练模型。为了计算类权重，我们是需要使用所有数据（训练、验证和测试）还是只使用训练集数据来计算类权重。谢谢

【参考方案1】：

在训练模型时，您只能假设训练数据可供您使用。 估计class_weights 是训练的一部分——它定义了你的损失函数。