如何在tensorflow数据集上使用样本权重？

我一直在使用 Tensorflow 和 Tensorflow Datasets 在 python 中训练用于多类语义分割的 unet 模型。

我注意到我的一个班级在培训中的代表性不足。在做了一些研究之后，我发现了样本权重，并认为这可能是我问题的一个很好的解决方案，但我一直无法破译有关如何使用它的文档或找到使用它的示例。

有人可以帮助解释样本权重如何与用于训练的数据集一起发挥作用，或者向我指出一个正在实施的示例吗？或者甚至 model.fit 函数期望的输入类型会有所帮助。