用于多类分类的 pytorch 类权重

Question

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

Answer 1

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