8.2.1　基于JAX的TensorFlow Datasets数据集分类实战