增加一些设置。
This commit is contained in:
@@ -111,7 +111,7 @@ builder = FAISSIndexBuilder(
|
||||
# 构建完整索引
|
||||
index = builder.build_complete_index(
|
||||
train_dir="../dataset/train",
|
||||
output_dir="faiss_index",
|
||||
output_dir="DishClassification/faiss_index",
|
||||
batch_size=16,
|
||||
index_type='flat' # 'flat', 'ivf', 'hnsw'
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user