Upload sd_token_similarity_calculator.ipynb

sd_token_similarity_calculator.ipynb

@@ -147,13 +147,23 @@
     {
       "cell_type": "code",
       "source": [
-        "list_size = 10 # @param {type:'number'}\n",
+        "list_size = 100 # @param {type:'number'}\n",
+        "\n",
+        "print_ID = False # @param {type:\"boolean\"}\n",
+        "print_Similarity = True # @param {type:\"boolean\"}\n",
+        "print_Name = True # @param {type:\"boolean\"}\n",
+        "print_Divider = False # @param {type:\"boolean\"}\n",
+        "\n",
         "for index in range(list_size):\n",
         "  id = indices[index].item()\n",
-        "  print(f'{vocab[id]}') # vocab item\n",
-        "  print(f'ID = {id}') # IDs\n",
-        "  print(f'similiarity = {round(sorted[index].item()*100,2)} %') # % value\n",
-        "  print('--------')\n"
+        "  if (print_Name):\n",
+        "    print(f'{vocab[id]}') # vocab item\n",
+        "  if (print_ID):\n",
+        "    print(f'ID = {id}') # IDs\n",
+        "  if (print_Similarity):\n",
+        "    print(f'similiarity = {round(sorted[index].item()*100,2)} %') # % value\n",
+        "  if (print_Divider):\n",
+        "    print('--------')"
       ],
       "metadata": {
         "id": "YIEmLAzbHeuo"