Changed arg_min(to be deprecated in future tensorflow versions) to argmin (aymericdamien#240)

BhavtoshRath · aymericdamien · commit f6f80833e025 · 2018-04-09T16:40:53.000-07:00
diff --git a/notebooks/2_BasicModels/nearest_neighbor.ipynb b/notebooks/2_BasicModels/nearest_neighbor.ipynb
@@ -62,7 +62,7 @@
     "# Calculate L1 Distance\n",
     "distance = tf.reduce_sum(tf.abs(tf.add(xtr, tf.negative(xte))), reduction_indices=1)\n",
     "# Prediction: Get min distance index (Nearest neighbor)\n",
-    "pred = tf.arg_min(distance, 0)\n",
+    "pred = tf.argmin(distance, 0)\n",
     "\n",
     "accuracy = 0.\n",
     "\n",
