wxy01giser commited on
Commit
992cb92
·
verified ·
1 Parent(s): 4b4a7f2

Update cluster_insight.py

Browse files
Files changed (1) hide show
  1. cluster_insight.py +1 -1
cluster_insight.py CHANGED
@@ -113,7 +113,7 @@ def encode_sentences_with_cache(sentences, model): #, cache_path='sentence_vecto
113
  # print("未找到缓存,开始编码...")
114
  sentence_vectors = model.encode(
115
  sentences,
116
- batch_size=16,
117
  show_progress_bar=False,
118
  convert_to_tensor=False,
119
  normalize_embeddings=True, # 直接归一化
 
113
  # print("未找到缓存,开始编码...")
114
  sentence_vectors = model.encode(
115
  sentences,
116
+ batch_size=32,
117
  show_progress_bar=False,
118
  convert_to_tensor=False,
119
  normalize_embeddings=True, # 直接归一化