增加了相似菜品预览功能,更完整了,交互更方便了。

This commit is contained in:
2025-09-29 18:03:28 +08:00
parent 1f04bb5c01
commit f3e3bd048a
4 changed files with 163 additions and 26 deletions
+1 -1
View File
@@ -477,7 +477,7 @@ class FAISSSearcher:
def main():
"""主函数"""
# 配置参数
MODEL_PATH = "../model/embedding_20250917_145342/best_embedding_model.pth"
MODEL_PATH = "../model/embedding_20250929_151102/best_embedding_model.pth"
TRAIN_DIR = "../dataset/train"
OUTPUT_DIR = "faiss_index"
BATCH_SIZE = 16