Merge pull request #701 from neelamgehlot/remove-tf-contrib-learn-dataset

Remove usage of tensorflow.contrib.learn since it is deprecated

Author: mvsusp

sagemaker-python-sdk/tensorflow_distributed_mnist/tensorflow_distributed_mnist.ipynb

@@ -51,10 +51,10 @@
    "outputs": [],
    "source": [
     "import utils\n",
-    "from tensorflow.contrib.learn.python.learn.datasets import mnist\n",
+    "from tensorflow.examples.tutorials.mnist import input_data\n",
     "import tensorflow as tf\n",
     "\n",
-    "data_sets = mnist.read_data_sets('data', dtype=tf.uint8, reshape=False, validation_size=5000)\n",
+    "data_sets = input_data.read_data_sets('data', dtype=tf.uint8, reshape=False, validation_size=5000)\n",
     "\n",
     "utils.convert_to(data_sets.train, 'train', 'data')\n",
     "utils.convert_to(data_sets.validation, 'validation', 'data')\n",

sagemaker-python-sdk/tensorflow_distributed_mnist/tensorflow_local_mode_mnist.ipynb

@@ -67,10 +67,10 @@
    "outputs": [],
    "source": [
     "import utils\n",
-    "from tensorflow.contrib.learn.python.learn.datasets import mnist\n",
+    "from tensorflow.examples.tutorials.mnist import input_data\n",
     "import tensorflow as tf\n",
     "\n",
-    "data_sets = mnist.read_data_sets('data', dtype=tf.uint8, reshape=False, validation_size=5000)\n",
+    "data_sets = input_data.read_data_sets('data', dtype=tf.uint8, reshape=False, validation_size=5000)\n",
     "\n",
     "utils.convert_to(data_sets.train, 'train', 'data')\n",
     "utils.convert_to(data_sets.validation, 'validation', 'data')\n",