如何使用.map（）函数张量流数据集修复“ KeyError”

Question

我一直在尝试为Rock剪刀纸数据集建立一个简单的模型。但是，当我尝试使用.map（）函数缩放数据时，出现TypeError。

IMAGE_RES = 300
def format_image(image, label):
  image = image/255.0
  return  image, label

train_batches = train_set.shuffle(2520//4).map(format_image).batch(BATCH_SIZE).prefetch(1)

validation_batches = test_set.map(format_image).batch(BATCH_SIZE).prefetch(1)

由于我在网上找到了类似的代码，所以我希望它能起作用，但是当我运行此代码时，会收到以下错误消息：

TypeError: format_image() missing 1 required positional argument: 'label'