소스 검색

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

Bhavtosh Rath 7 년 전
부모
커밋
f6f80833e0
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  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",