diff --git a/src/linalg/mod.rs b/src/linalg/mod.rs index 4fb259f..c560b78 100644 --- a/src/linalg/mod.rs +++ b/src/linalg/mod.rs @@ -86,11 +86,11 @@ pub trait BaseVector: Clone + Debug { /// Create a new vector from a &[T] /// ``` /// use smartcore::linalg::naive::dense_matrix::*; - /// let slice: &[f64] = &[0., 0.5, 2., 3., 4.]; - /// let a: Vec = BaseVector::from_slice(slice); - /// assert_eq!(a, vec![0., 0.5, 2., 3., 4.]); + /// let a: [f64; 5] = [0., 0.5, 2., 3., 4.]; + /// let v: Vec = BaseVector::from_array(&a); + /// assert_eq!(v, vec![0., 0.5, 2., 3., 4.]); /// ``` - fn from_slice(f: &[T]) -> Self { + fn from_array(f: &[T]) -> Self { let mut v = Self::zeros(f.len()); for (i, elem) in f.iter().enumerate() { v.set(i, *elem); diff --git a/src/naive_bayes/mod.rs b/src/naive_bayes/mod.rs index ffc3e2e..f93d3bf 100644 --- a/src/naive_bayes/mod.rs +++ b/src/naive_bayes/mod.rs @@ -58,7 +58,7 @@ impl, D: NBDistribution> BaseNaiveBayes>(); - let y_hat = M::RowVector::from_slice(&predictions); + let y_hat = M::RowVector::from_array(&predictions); Ok(y_hat) } }