chouchouvs commited on
Commit
6eb5a6e
·
verified ·
1 Parent(s): 781b76e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -268,7 +268,7 @@ def _save_faiss(fx_dir: str, xb: np.ndarray, meta: Dict[str, Any]) -> None:
268
 
269
  # entraînement sur un sous‑échantillon (max 10 k vecteurs)
270
  rng = np.random.default_rng(0)
271
- train = xb[rng.choice(xb.shape[0], min(10_000, xb.shape[0]), replace=False]
272
  index.train(train)
273
 
274
  index.add(xb)
 
268
 
269
  # entraînement sur un sous‑échantillon (max 10 k vecteurs)
270
  rng = np.random.default_rng(0)
271
+ train = xb[rng.choice(xb.shape[0], min(10_000, xb.shape[0]), replace=False)]
272
  index.train(train)
273
 
274
  index.add(xb)