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

reset

Browse Source
This commit is contained in:
zhangyiqun01
2021-02-26 10:47:21 +08:00
parent 0e9c517b1a
commit cd44f1d515
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/svm/svc.rs
+6 -6
View File
@@ -58,7 +58,7 @@
//! let y = vec![ 0., 0., 0., 0., 0., 0., 0., 0., //! let y = vec![ 0., 0., 0., 0., 0., 0., 0., 0.,
//! 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 svc = SVC::fit(&x, &y, //! let svr = SVC::fit(&x, &y,
//! Kernels::linear(), //! Kernels::linear(),
//! SVCParameters { //! SVCParameters {
//! epoch: 2, //! epoch: 2,
@@ -66,7 +66,7 @@
//! tol: 1e-3, //! tol: 1e-3,
//! }).unwrap(); //! }).unwrap();
//! //!
//! let y_hat = svc.predict(&x).unwrap(); //! let y_hat = svr.predict(&x).unwrap();
//! ``` //! ```
//! //!
//! ## References: //! ## References:
@@ -802,11 +802,11 @@ mod tests {
-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 svc = SVC::fit(&x, &y, Kernels::linear(), Default::default()).unwrap(); let svr = SVC::fit(&x, &y, Kernels::linear(), Default::default()).unwrap();
let deserialized_svc: SVC<f64, DenseMatrix<f64>, LinearKernel> = let deserialized_svr: SVC<f64, DenseMatrix<f64>, LinearKernel> =
serde_json::from_str(&serde_json::to_string(&svc).unwrap()).unwrap(); serde_json::from_str(&serde_json::to_string(&svr).unwrap()).unwrap();
assert_eq!(svc, deserialized_svc); assert_eq!(svr, deserialized_svr);
} }
} }