update coefficients

This commit is contained in:
Luc Mcgrady 2025-06-05 23:49:23 +01:00
parent b37fb4fb93
commit eb38c7014b
No known key found for this signature in database
GPG key ID: 4F3D7A0B17CC3D9C

View file

@ -63,14 +63,14 @@ pub struct ComputeParamsRequest<'t> {
/// r: retention /// r: retention
fn log_loss_adjustment(r: f32) -> f32 { fn log_loss_adjustment(r: f32) -> f32 {
0.621 * (4. * r * (1. - r)).powf(0.739) 0.623 * (4. * r * (1. - r)).powf(0.738)
} }
/// r: retention /// r: retention
/// ///
/// c: review count /// c: review count
fn rmse_adjustment(r: f32, c: u32) -> f32 { fn rmse_adjustment(r: f32, c: u32) -> f32 {
0.0417 / (r.powf(1.63) - 1.41) + 0.125 / ((c as f32 / 1000.).powf(0.655) + 1.22) + 0.102 0.0135 / (r.powf(0.504) - 1.14) + 0.176 / ((c as f32 / 1000.).powf(0.825) + 2.22) + 0.101
} }
impl Collection { impl Collection {