Reordered methods [skip ci]
This commit is contained in:
16
src/model.rs
16
src/model.rs
@@ -54,14 +54,6 @@ impl Model {
|
|||||||
unsafe { (*self.model).b }
|
unsafe { (*self.model).b }
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn p_iter(&self) -> Chunks<'_, f32> {
|
|
||||||
self.p_factors().chunks(self.factors() as usize)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn q_iter(&self) -> Chunks<'_, f32> {
|
|
||||||
self.q_factors().chunks(self.factors() as usize)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn p_factors(&self) -> &[f32] {
|
pub fn p_factors(&self) -> &[f32] {
|
||||||
unsafe { std::slice::from_raw_parts((*self.model).p, (self.rows() * self.factors()) as usize) }
|
unsafe { std::slice::from_raw_parts((*self.model).p, (self.rows() * self.factors()) as usize) }
|
||||||
}
|
}
|
||||||
@@ -90,6 +82,14 @@ impl Model {
|
|||||||
None
|
None
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn p_iter(&self) -> Chunks<'_, f32> {
|
||||||
|
self.p_factors().chunks(self.factors() as usize)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn q_iter(&self) -> Chunks<'_, f32> {
|
||||||
|
self.q_factors().chunks(self.factors() as usize)
|
||||||
|
}
|
||||||
|
|
||||||
pub fn rmse(&self, data: &Matrix) -> f64 {
|
pub fn rmse(&self, data: &Matrix) -> f64 {
|
||||||
let prob = data.to_problem();
|
let prob = data.to_problem();
|
||||||
unsafe { calc_rmse(&prob, self.model) }
|
unsafe { calc_rmse(&prob, self.model) }
|
||||||
|
|||||||
Reference in New Issue
Block a user