dymk/libmf-rust
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updated docs [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2024-06-30 22:40:09 -07:00
parent 9066cb543d
commit 877e87a501
2 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
README.md
View File

@@ -90,10 +90,10 @@ libmf::Model::params()
.threads(12) // number of threads
.bins(25) // number of bins
.iterations(20) // number of iterations
.lambda_p1(0.0) // coefficient of L1-norm regularization on P
.lambda_p2(0.1) // coefficient of L2-norm regularization on P
.lambda_q1(0.0) // coefficient of L1-norm regularization on Q
.lambda_q2(0.1) // coefficient of L2-norm regularization on Q
.lambda_p1(0.0) // coefficient of L1-norm regularization for P
.lambda_p2(0.1) // coefficient of L2-norm regularization for P
.lambda_q1(0.0) // coefficient of L1-norm regularization for Q
.lambda_q2(0.1) // coefficient of L2-norm regularization for Q
.learning_rate(0.1) // learning rate
.alpha(1.0) // importance of negative entries
.c(0.0001) // desired value of negative entries

8
src/params.rs
View File

@@ -43,25 +43,25 @@ impl Params {
self
}
/// Sets the coefficient of L1-norm regularization on P.
/// Sets the coefficient of L1-norm regularization for P.
pub fn lambda_p1(&mut self, value: f32) -> &mut Self {
self.param.lambda_p1 = value;
self
}
/// Sets the coefficient of L2-norm regularization on P.
/// Sets the coefficient of L2-norm regularization for P.
pub fn lambda_p2(&mut self, value: f32) -> &mut Self {
self.param.lambda_p2 = value;
self
}
/// Sets the coefficient of L1-norm regularization on Q.
/// Sets the coefficient of L1-norm regularization for Q.
pub fn lambda_q1(&mut self, value: f32) -> &mut Self {
self.param.lambda_q1 = value;
self
}
/// Sets the coefficient of L2-norm regularization on Q.
/// Sets the coefficient of L2-norm regularization for Q.
pub fn lambda_q2(&mut self, value: f32) -> &mut Self {
self.param.lambda_q2 = value;
self