Merge pull request #44764 from gbaned:master

PiperOrigin-RevId: 342087238 Change-Id: I3d22a2ee03a288156a9e6dd72f80fb9055003fa3
tensorflow · Nov 12, 2020 · 1028958 · 1028958
2 parents 69a1404 + e39f63d
commit 1028958
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflow/python/keras/preprocessing/image.py b/tensorflow/python/keras/preprocessing/image.py
@@ -661,8 +661,8 @@ class ImageDataGenerator(image.ImageDataGenerator):
 
   ```python
   (x_train, y_train), (x_test, y_test) = cifar10.load_data()
-  y_train = np_utils.to_categorical(y_train, num_classes)
-  y_test = np_utils.to_categorical(y_test, num_classes)
+  y_train = utils.to_categorical(y_train, num_classes)
+  y_test = utils.to_categorical(y_test, num_classes)
   datagen = ImageDataGenerator(
       featurewise_center=True,
       featurewise_std_normalization=True,