{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "from utils import *\n",
    "warnings.filterwarnings('ignore')\n",
    "%matplotlib inline"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "fd_collection = getCollection(\"team_5_mwdb_phase_2\", \"fd_collection\")\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Applying svd on the cm_fd space to get 10 latent semantics (showing only top 10 image-weight pairs for each latent semantic)...\n",
      "Latent semantic no. 0\n",
      "Image_ID\t7654\t-\tWeight\t0.08162189274964751\n",
      "Image_ID\t8634\t-\tWeight\t0.06673589485778451\n",
      "Image_ID\t5740\t-\tWeight\t0.060058821201972104\n",
      "Image_ID\t6106\t-\tWeight\t0.05306661393931607\n",
      "Image_ID\t5456\t-\tWeight\t0.05170171570330845\n",
      "Image_ID\t7814\t-\tWeight\t0.04997978865116185\n",
      "Image_ID\t6248\t-\tWeight\t0.04946683639815072\n",
      "Image_ID\t5354\t-\tWeight\t0.04864381025793171\n",
      "Image_ID\t6108\t-\tWeight\t0.04796763934338538\n",
      "Image_ID\t5438\t-\tWeight\t0.047874747600689466\n",
      "Latent semantic no. 1\n",
      "Image_ID\t8026\t-\tWeight\t0.06478360955460367\n",
      "Image_ID\t6016\t-\tWeight\t0.0632709906607753\n",
      "Image_ID\t3744\t-\tWeight\t0.05347414608321652\n",
      "Image_ID\t3720\t-\tWeight\t0.0517124023583583\n",
      "Image_ID\t7896\t-\tWeight\t0.049366978424645006\n",
      "Image_ID\t6014\t-\tWeight\t0.047637173390389816\n",
      "Image_ID\t6768\t-\tWeight\t0.04742408995375774\n",
      "Image_ID\t4050\t-\tWeight\t0.0456343920101654\n",
      "Image_ID\t6000\t-\tWeight\t0.04535273415975713\n",
      "Image_ID\t6552\t-\tWeight\t0.04525300117499444\n",
      "Latent semantic no. 2\n",
      "Image_ID\t7654\t-\tWeight\t0.0704670166327785\n",
      "Image_ID\t2804\t-\tWeight\t0.059682344110996065\n",
      "Image_ID\t2710\t-\tWeight\t0.059199111598090534\n",
      "Image_ID\t3436\t-\tWeight\t0.05368202357324355\n",
      "Image_ID\t7936\t-\tWeight\t0.053276991496894154\n",
      "Image_ID\t2708\t-\tWeight\t0.048527019795007204\n",
      "Image_ID\t3764\t-\tWeight\t0.04835537239641643\n",
      "Image_ID\t7928\t-\tWeight\t0.047998989024259496\n",
      "Image_ID\t5684\t-\tWeight\t0.04723047448150771\n",
      "Image_ID\t5126\t-\tWeight\t0.04720498270016634\n",
      "Latent semantic no. 3\n",
      "Image_ID\t6356\t-\tWeight\t0.0754447261688377\n",
      "Image_ID\t6480\t-\tWeight\t0.06540890240964665\n",
      "Image_ID\t4756\t-\tWeight\t0.06075370676621832\n",
      "Image_ID\t8656\t-\tWeight\t0.060505116069252685\n",
      "Image_ID\t6050\t-\tWeight\t0.058111632773274836\n",
      "Image_ID\t6324\t-\tWeight\t0.056492568599917435\n",
      "Image_ID\t8138\t-\tWeight\t0.0557967464751822\n",
      "Image_ID\t3460\t-\tWeight\t0.05508818833516222\n",
      "Image_ID\t200\t-\tWeight\t0.05459477384213874\n",
      "Image_ID\t7220\t-\tWeight\t0.05376222500332758\n",
      "Latent semantic no. 4\n",
      "Image_ID\t7370\t-\tWeight\t0.05281026462493995\n",
      "Image_ID\t6528\t-\tWeight\t0.05252803707219332\n",
      "Image_ID\t8056\t-\tWeight\t0.05175019567880743\n",
      "Image_ID\t2958\t-\tWeight\t0.05123118911737749\n",
      "Image_ID\t4614\t-\tWeight\t0.05061302210733273\n",
      "Image_ID\t8292\t-\tWeight\t0.05000577057549489\n",
      "Image_ID\t7888\t-\tWeight\t0.04905059301012787\n",
      "Image_ID\t6540\t-\tWeight\t0.048139958875035395\n",
      "Image_ID\t6064\t-\tWeight\t0.04605896293857696\n",
      "Image_ID\t2974\t-\tWeight\t0.04488429099909397\n",
      "Latent semantic no. 5\n",
      "Image_ID\t8570\t-\tWeight\t0.08379938013632145\n",
      "Image_ID\t7784\t-\tWeight\t0.0723847258804912\n",
      "Image_ID\t4152\t-\tWeight\t0.060769224719766333\n",
      "Image_ID\t5114\t-\tWeight\t0.053872121517690504\n",
      "Image_ID\t7774\t-\tWeight\t0.05324887247523992\n",
      "Image_ID\t8614\t-\tWeight\t0.05319742868629013\n",
      "Image_ID\t3072\t-\tWeight\t0.05083994521792821\n",
      "Image_ID\t7798\t-\tWeight\t0.05059807413594892\n",
      "Image_ID\t5118\t-\tWeight\t0.05022770477320976\n",
      "Image_ID\t7040\t-\tWeight\t0.04996996742218053\n",
      "Latent semantic no. 6\n",
      "Image_ID\t8570\t-\tWeight\t0.07082421149695754\n",
      "Image_ID\t7774\t-\tWeight\t0.06546594547486781\n",
      "Image_ID\t4152\t-\tWeight\t0.06440870014673936\n",
      "Image_ID\t5118\t-\tWeight\t0.06264436903974217\n",
      "Image_ID\t7784\t-\tWeight\t0.06203552824772956\n",
      "Image_ID\t7798\t-\tWeight\t0.05899354962287134\n",
      "Image_ID\t7896\t-\tWeight\t0.05648444493570963\n",
      "Image_ID\t7766\t-\tWeight\t0.056063042928801675\n",
      "Image_ID\t7792\t-\tWeight\t0.055578803018497686\n",
      "Image_ID\t7834\t-\tWeight\t0.055567509183302555\n",
      "Latent semantic no. 7\n",
      "Image_ID\t7912\t-\tWeight\t0.06634864556518678\n",
      "Image_ID\t5534\t-\tWeight\t0.05913926717735747\n",
      "Image_ID\t5550\t-\tWeight\t0.049468125695492526\n",
      "Image_ID\t2106\t-\tWeight\t0.048274676516220805\n",
      "Image_ID\t7804\t-\tWeight\t0.04822832951751611\n",
      "Image_ID\t6198\t-\tWeight\t0.04795521082538372\n",
      "Image_ID\t6728\t-\tWeight\t0.04729135404469566\n",
      "Image_ID\t5588\t-\tWeight\t0.04715637083533252\n",
      "Image_ID\t7276\t-\tWeight\t0.04637482601331893\n",
      "Image_ID\t6730\t-\tWeight\t0.045930617636659\n",
      "Latent semantic no. 8\n",
      "Image_ID\t1798\t-\tWeight\t0.04586412291217343\n",
      "Image_ID\t1802\t-\tWeight\t0.044772142290101236\n",
      "Image_ID\t1806\t-\tWeight\t0.044448676280621977\n",
      "Image_ID\t1202\t-\tWeight\t0.043679466488681935\n",
      "Image_ID\t1786\t-\tWeight\t0.04351371229636818\n",
      "Image_ID\t1784\t-\tWeight\t0.04346765741634348\n",
      "Image_ID\t1790\t-\tWeight\t0.04288750664761761\n",
      "Image_ID\t1642\t-\tWeight\t0.041863484069841805\n",
      "Image_ID\t1788\t-\tWeight\t0.04089406629514228\n",
      "Image_ID\t1796\t-\tWeight\t0.04068815222347919\n",
      "Latent semantic no. 9\n",
      "Image_ID\t8582\t-\tWeight\t0.02577153311253718\n",
      "Image_ID\t8612\t-\tWeight\t0.025608143819276445\n",
      "Image_ID\t7290\t-\tWeight\t0.025578071187110543\n",
      "Image_ID\t7298\t-\tWeight\t0.025350467801040884\n",
      "Image_ID\t7302\t-\tWeight\t0.02531661140938117\n",
      "Image_ID\t7318\t-\tWeight\t0.025212779767014252\n",
      "Image_ID\t8580\t-\tWeight\t0.025201323062899284\n",
      "Image_ID\t6392\t-\tWeight\t0.02517086205642468\n",
      "Image_ID\t2738\t-\tWeight\t0.025106516897995135\n",
      "Image_ID\t6420\t-\tWeight\t0.02510499876667641\n"
     ]
    }
   ],
   "source": [
    "selected_feature_model = valid_feature_models[\n",
    "    str(input(\"Enter feature model - one of \" + str(list(valid_feature_models.keys()))))\n",
    "]\n",
    "\n",
    "k = int(input(\"Enter value of k: \"))\n",
    "if k < 1:\n",
    "    raise ValueError(\"k should be a positive integer\")\n",
    "\n",
    "selected_dim_reduction_method = str(\n",
    "    input(\n",
    "        \"Enter dimensionality reduction method - one of \"\n",
    "        + str(list(valid_dim_reduction_methods.keys()))\n",
    "    )\n",
    ")\n",
    "\n",
    "extract_latent_semantics(\n",
    "    fd_collection,\n",
    "    k,\n",
    "    selected_feature_model,\n",
    "    selected_dim_reduction_method,\n",
    "    top_images=10,\n",
    ")\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}