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

make default params available to serde (#167)

Browse Source 
* add seed param to search params

* make default params available to serde

* lints

* create defaults for enums

* lint
This commit is contained in:
Montana Low
2022-09-21 19:48:31 -07:00
committed by GitHub
parent 403d3f2348
commit 764309e313
22 changed files with 175 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/svm/svc.rs
+12
View File
@@ -91,16 +91,22 @@ use crate::svm::{Kernel, Kernels, LinearKernel};
#[derive(Debug, Clone)]
/// SVC Parameters
pub struct SVCParameters<T: RealNumber, M: Matrix<T>, K: Kernel<T, M::RowVector>> {
#[cfg_attr(feature = "serde", serde(default))]
/// Number of epochs.
pub epoch: usize,
#[cfg_attr(feature = "serde", serde(default))]
/// Regularization parameter.
pub c: T,
#[cfg_attr(feature = "serde", serde(default))]
/// Tolerance for stopping criterion.
pub tol: T,
#[cfg_attr(feature = "serde", serde(default))]
/// The kernel function.
pub kernel: K,
#[cfg_attr(feature = "serde", serde(default))]
/// Unused parameter.
m: PhantomData<M>,
#[cfg_attr(feature = "serde", serde(default))]
/// Controls the pseudo random number generation for shuffling the data for probability estimates
seed: Option<u64>,
}
@@ -109,16 +115,22 @@ pub struct SVCParameters<T: RealNumber, M: Matrix<T>, K: Kernel<T, M::RowVector>
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
#[derive(Debug, Clone)]
pub struct SVCSearchParameters<T: RealNumber, M: Matrix<T>, K: Kernel<T, M::RowVector>> {
#[cfg_attr(feature = "serde", serde(default))]
/// Number of epochs.
pub epoch: Vec<usize>,
#[cfg_attr(feature = "serde", serde(default))]
/// Regularization parameter.
pub c: Vec<T>,
#[cfg_attr(feature = "serde", serde(default))]
/// Tolerance for stopping epoch.
pub tol: Vec<T>,
#[cfg_attr(feature = "serde", serde(default))]
/// The kernel function.
pub kernel: Vec<K>,
#[cfg_attr(feature = "serde", serde(default))]
/// Unused parameter.
m: PhantomData<M>,
#[cfg_attr(feature = "serde", serde(default))]
/// Controls the pseudo random number generation for shuffling the data for probability estimates
seed: Vec<Option<u64>>,
}