问题描述:
训练数据分布不均匀,需要对类别施加权重——class_weight,该如何改写mindspore.nn.SoftmaxCrossEntropyWithLogits呢,希望可以给出示例代码
解决方案:
可以参考以下教程进行自定义损失函数:
https://www.mindspore.cn/tutorial/training/zh-CN/master/advanced_use/custom_loss_function.html
损失函数的定义方式是通用的,本质上就是一个Cell,在其中实现需要的算法就可以了。