Home Explore Help
Sign In
radu
/
TensorFlow-Examples
mirror of https://github.com/aymericdamien/TensorFlow-Examples.git
Watch 1
Star 0
Fork 0
Files Issues 0 Wiki
Browse Source

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

Bhavtosh Rath 7 years ago
parent
e43556dfc0
commit
f6f80833e0
1 changed files with 1 additions and 1 deletions
Unified View Show Diff Stats
  1. 1 1
      notebooks/2_BasicModels/nearest_neighbor.ipynb

+ 1 - 1
notebooks/2_BasicModels/nearest_neighbor.ipynb
View File 

@@ -62,7 +62,7 @@
 
     "# Calculate L1 Distance\n",
 
     "# Calculate L1 Distance\n",
 
     "distance = tf.reduce_sum(tf.abs(tf.add(xtr, tf.negative(xte))), reduction_indices=1)\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",
 
     "# Prediction: Get min distance index (Nearest neighbor)\n",
 
-    "pred = tf.arg_min(distance, 0)\n",
 
+    "pred = tf.argmin(distance, 0)\n",
 
     "\n",
 
     "\n",
 
     "accuracy = 0.\n",
 
     "accuracy = 0.\n",
 
     "\n",
 
     "\n",