Skip to content

Commit f6f8083

Browse files
BhavtoshRathaymericdamien
authored andcommitted
Changed arg_min(to be deprecated in future tensorflow versions) to argmin (aymericdamien#240)
1 parent e43556d commit f6f8083

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

notebooks/2_BasicModels/nearest_neighbor.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@
6262
"# Calculate L1 Distance\n",
6363
"distance = tf.reduce_sum(tf.abs(tf.add(xtr, tf.negative(xte))), reduction_indices=1)\n",
6464
"# Prediction: Get min distance index (Nearest neighbor)\n",
65-
"pred = tf.arg_min(distance, 0)\n",
65+
"pred = tf.argmin(distance, 0)\n",
6666
"\n",
6767
"accuracy = 0.\n",
6868
"\n",

0 commit comments

Comments
 (0)