Hi, I’m new to TensorFlow. While applying tutorial code from ’ Using tf.data for finer control’ section of the ‘Load and Preprocess data’ tutorial to my own data that have two categories, I obtained error messages apparently related to loss functions. The two loss functions below compile but produce errors during model fitting:
SparseCategoricalCrossentropy results in ‘InvalidArgumentError: Received a label value of 2 which is outside the valid range of [0, 2).’ This is the loss function used in the tutorial.
BinaryCrossentropy results in ‘ValueError: logits and labels must have the same shape ((None, 2) vs (None, 1))’. I tried this as an alternative.
I should also note that I was able to successfully fit a model to my data using the methods described in the first part of the tutorial. It’s only the approach used in the second part of the tutorial that raises errors.