增加一些设置。

2025-10-17 18:24:04 +08:00
parent 25f8e0a267
commit 5db96600fe
10 changed files with 38 additions and 25 deletions
@@ -470,7 +470,7 @@ def main():
    # 配置参数
    MODEL_PATH = "model/embedding_20250930_102826/best_embedding_model.pth"
    TRAIN_DIR = "dataset/train"
-    OUTPUT_DIR = "faiss_vector_db/faiss_index"
+    OUTPUT_DIR = "faiss_vector_db/DishClassification/faiss_index"
    BATCH_SIZE = 16
    INDEX_TYPE = 'flat'  # 'flat', 'ivf', 'hnsw'
    EMBEDDING_DIM = 512