This commit is contained in:
Niraj Sonje 2023-10-14 17:59:55 -07:00
parent b657e75c42
commit ec77eb4d7c

View File

@ -158,6 +158,15 @@
" else:\n",
" comparison_vector = np.matmul(label_rep, np.transpose(H))\n",
"\n",
" case \"lda\":\n",
" comparison_feature_space = np.array(data[\"image-semantic\"])\n",
" label_vectors = []\n",
" length = len(comparison_feature_space)\n",
" for i in range(length):\n",
" if all_images[i][\"true_label\"] == label:\n",
" label_vectors.append(comparison_feature_space[i])\n",
" comparison_vector = [sum(col) / len(col) for col in zip(*label_vectors)] \n",
" \n",
" case \"kmeans\":\n",
" comparison_vector = []\n",
" comparison_feature_space = np.array(data[\"image-semantic\"])\n",
@ -240,6 +249,10 @@
" comparison_feature_space = np.array(data['image-semantic'])\n",
" comparison_vector = comparison_feature_space[label]\n",
"\n",
" case \"lda\":\n",
" comparison_feature_space = np.array(data['image-semantic'])\n",
" comparison_vector = comparison_feature_space[label]\n",
"\n",
" case \"kmeans\":\n",
" comparison_feature_space = np.array(data[\"image-semantic\"])\n",
" comparison_vector = comparison_feature_space[label]\n",