title: How to use sample weights with tensorflow datasets?
body: <p>I have been training a unet model for multiclass semantic segmentation in python using Tensorflow and Tensorflow Datasets.</p>
  <p>I've noticed that one of my classes seems to be underrepresented in training. After doing some research, I found out about sample weights and thought it might be a good solution to my problem, but I've been having trouble deciphering the documentation on how to use it or find examples of it being used.</p>
  <p>Could someone help explain how sample weights come into play with datasets for training or point me to an example where it is being implemented? Or even what type of input the model.fit function is expecting would be helpful.</p>