Parcourir la source

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

Bhavtosh Rath il y a 7 ans
Parent
commit
f6f80833e0
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      notebooks/2_BasicModels/nearest_neighbor.ipynb

+ 1 - 1
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",