Update app.py
Browse files
app.py
CHANGED
@@ -109,10 +109,13 @@ def filter_semantically_similar_texts_by_embedding(df, mode, embedding_field='em
|
|
109 |
|
110 |
def search_kpi(kpi_query, kpi_count, mode):
|
111 |
if mode == "BGE":
|
|
|
112 |
results = kpi_pool_ori.similarity_search_with_relevance_scores(kpi_query, k=50)
|
113 |
elif mode == "SBERT-snunlp":
|
|
|
114 |
results = kpi_pool.similarity_search_with_relevance_scores(kpi_query, k=50)
|
115 |
else:
|
|
|
116 |
results = kpi_pool2.similarity_search_with_relevance_scores(kpi_query, k=50)
|
117 |
|
118 |
|
@@ -142,7 +145,7 @@ def search_kpi_all_models(kpi_query, kpi_count):
|
|
142 |
_, _, df_bge, _, _, _ = search_kpi(kpi_query, kpi_count, "BGE")
|
143 |
_, _, df_sn, _, _, _ = search_kpi(kpi_query, kpi_count, "SBERT-snunlp")
|
144 |
_, _, df_jh, _, _, _ = search_kpi(kpi_query, kpi_count, "SBERT-jhgan")
|
145 |
-
|
146 |
return df_bge, df_sn, df_jh
|
147 |
|
148 |
|
|
|
109 |
|
110 |
def search_kpi(kpi_query, kpi_count, mode):
|
111 |
if mode == "BGE":
|
112 |
+
print("BGE 검색 시작")
|
113 |
results = kpi_pool_ori.similarity_search_with_relevance_scores(kpi_query, k=50)
|
114 |
elif mode == "SBERT-snunlp":
|
115 |
+
print("SBERT-snunlp 검색 시작")
|
116 |
results = kpi_pool.similarity_search_with_relevance_scores(kpi_query, k=50)
|
117 |
else:
|
118 |
+
print("SBERT-jhgan 검색 시작")
|
119 |
results = kpi_pool2.similarity_search_with_relevance_scores(kpi_query, k=50)
|
120 |
|
121 |
|
|
|
145 |
_, _, df_bge, _, _, _ = search_kpi(kpi_query, kpi_count, "BGE")
|
146 |
_, _, df_sn, _, _, _ = search_kpi(kpi_query, kpi_count, "SBERT-snunlp")
|
147 |
_, _, df_jh, _, _, _ = search_kpi(kpi_query, kpi_count, "SBERT-jhgan")
|
148 |
+
print("함수 종료")
|
149 |
return df_bge, df_sn, df_jh
|
150 |
|
151 |
|