optimize with capacity

chengmiaomiao.123 · miamia0 · commit 4c73d7914f36 · 2025-02-21T14:50:50.000+08:00
diff --git a/src/hnsw/hnsw_const.rs b/src/hnsw/hnsw_const.rs
@@ -64,6 +64,17 @@ where
             params,
         }
     }
+
+    pub fn new_with_capacity(metric: Met, params: Params, capacity: usize) -> Self {
+        Self {
+            metric,
+            zero: Vec::with_capacity(capacity),
+            features: Vec::with_capacity(capacity),
+            layers: vec![],
+            prng: R::from_seed(R::Seed::default()),
+            params,
+        }
+    }
 }
 
 impl<Met, T, R, const M: usize, const M0: usize> Knn for Hnsw<Met, T, R, M, M0>

Original file line number	Diff line number	Diff line change
`@@ -64,6 +64,17 @@ where`
`64`	`64`	`params,`
`65`	`65`	`}`
`66`	`66`	`}`
	`67`	`+`
	`68`	`+ pub fn new_with_capacity(metric: Met, params: Params, capacity: usize) -> Self {`
	`69`	`+ Self {`
	`70`	`+ metric,`
	`71`	`+ zero: Vec::with_capacity(capacity),`
	`72`	`+ features: Vec::with_capacity(capacity),`
	`73`	`+ layers: vec![],`
	`74`	`+ prng: R::from_seed(R::Seed::default()),`
	`75`	`+ params,`
	`76`	`+ }`
	`77`	`+ }`
`67`	`78`	`}`
`68`	`79`
`69`	`80`	`impl<Met, T, R, const M: usize, const M0: usize> Knn for Hnsw<Met, T, R, M, M0>`