From 797dc3c8e08c221f1978a231fd04d2ba07d7f552 Mon Sep 17 00:00:00 2001
From: Volodymyr Orlov <vorl@Echo.local>
Date: Wed, 28 Oct 2020 17:23:40 -0700
Subject: [PATCH] fix: formatting

---
 src/svm/mod.rs |  2 +-
 src/svm/svc.rs | 12 +++++++-----
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/svm/mod.rs b/src/svm/mod.rs
index 0605907..97955e7 100644
--- a/src/svm/mod.rs
+++ b/src/svm/mod.rs
@@ -79,7 +79,7 @@ impl Kernels {
 #[derive(Serialize, Deserialize, Debug)]
 pub struct LinearKernel {}
 
-/// Radial basis function (Gaussian) kernel 
+/// Radial basis function (Gaussian) kernel
 pub struct RBFKernel<T: RealNumber> {
     /// kernel coefficient
     pub gamma: T,
diff --git a/src/svm/svc.rs b/src/svm/svc.rs
index 46625a9..97347fd 100644
--- a/src/svm/svc.rs
+++ b/src/svm/svc.rs
@@ -155,7 +155,8 @@ impl<T: RealNumber, M: Matrix<T>, K: Kernel<T, M::RowVector>> SVC<T, M, K> {
 
         if classes.len() != 2 {
             return Err(Failed::fit(&format!(
-                "Incorrect number of classes {}", classes.len()
+                "Incorrect number of classes {}",
+                classes.len()
             )));
         }
 
@@ -166,7 +167,7 @@ impl<T: RealNumber, M: Matrix<T>, K: Kernel<T, M::RowVector>> SVC<T, M, K> {
             if y_v != -T::one() || y_v != T::one() {
                 match y_v == classes[0] {
                     true => y.set(i, -T::one()),
-                    false => y.set(i, T::one())
+                    false => y.set(i, T::one()),
                 }
             }
         }
@@ -194,7 +195,7 @@ impl<T: RealNumber, M: Matrix<T>, K: Kernel<T, M::RowVector>> SVC<T, M, K> {
         for i in 0..n {
             let cls_idx = match self.predict_for_row(x.get_row(i)) == T::one() {
                 false => self.classes[0],
-                true => self.classes[1]
+                true => self.classes[1],
             };
             y_hat.set(i, cls_idx);
         }
@@ -720,7 +721,8 @@ mod tests {
         ]);
 
         let y: Vec<f64> = vec![
-            -1., -1., -1., -1., -1., -1., -1., -1., -1., -1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
+            -1., -1., -1., -1., -1., -1., -1., -1., -1., -1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
+            1.,
         ];
 
         let y_hat = SVC::fit(
@@ -734,7 +736,7 @@ mod tests {
             },
         )
         .and_then(|lr| lr.predict(&x))
-        .unwrap();        
+        .unwrap();
 
         assert!(accuracy(&y_hat, &y) >= 0.9);
     }