|
@@ -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",
|