python openai_access/extract_mrc_knn.py

[hc971545129]

python openai_access/extract_mrc_knn.py
06/20/2024 14:23:22 - INFO - faiss.loader - Loading faiss with AVX2 support.
06/20/2024 14:23:22 - INFO - faiss.loader - Successfully loaded faiss with AVX2 support.
100%|???????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????| 37/37 [00:23<00:00, 1.59it/s]
100%|???????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????| 37/37 [00:23<00:00, 1.59it/s]
100%|???????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????| 37/37 [00:23<00:00, 1.56it/s]
100%|????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????| 1/1 [00:00<00:00, 129.21it/s]
Traceback (most recent call last):
File "openai_access/extract_mrc_knn.py", line 233, in
index_, value_ = compute_simcse_knn(test_mrc_data=test_mrc_data, train_mrc_data=train_mrc_data, knn_num=32)
File "openai_access/extract_mrc_knn.py", line 101, in compute_simcse_knn
top_value, top_index = train_index[label].search(embedding.astype(np.float32), knn_num)
KeyError: 'ORG'