work/smartcore
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: adds KMeans clustering algorithm

Browse Source
This commit is contained in:
Volodymyr Orlov
2020-02-20 18:43:24 -08:00
parent 4359d66bfa
commit 0e89113297
13 changed files with 637 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/linalg/ndarray_bindings.rs
-21
View File
@@ -16,14 +16,6 @@ impl Matrix for ArrayBase<OwnedRepr<f64>, Ix2>
self.into_shape(vec_size).unwrap()
}
fn from_array(nrows: usize, ncols: usize, values: &[f64]) -> Self {
Array::from_shape_vec((nrows, ncols), values.to_vec()).unwrap()
}
fn from_vec(nrows: usize, ncols: usize, values: Vec<f64>) -> Self {
Array::from_shape_vec((nrows, ncols), values).unwrap()
}
fn get(&self, row: usize, col: usize) -> f64 {
self[[row, col]]
}
@@ -330,19 +322,6 @@ mod tests {
}
#[test]
fn from_array_from_vec() {
let a1 = arr2(&[[ 1., 2., 3.],
[4., 5., 6.]]);
let a2 = Array2::from_array(2, 3, &[1., 2., 3., 4., 5., 6.]);
let a3 = Array2::from_vec(2, 3, vec![1., 2., 3., 4., 5., 6.]);
assert_eq!(a1, a2);
assert_eq!(a1, a3);
}
#[test]
fn vstack_hstack() {