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

Fixed name

Browse Source
This commit is contained in:
Andrew Kane
2021-07-26 14:13:32 -07:00
parent 7b4900afb1
commit 27cb6fa2a9
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/bindings.rs
View File

@@ -63,6 +63,6 @@ extern "C" {
pub fn calc_gkl(prob: *const MfProblem, model: *const MfModel) -> c_double;
pub fn calc_logloss(prob: *const MfProblem, model: *const MfModel) -> c_double;
pub fn calc_accuracy(prob: *const MfProblem, model: *const MfModel) -> c_double;
pub fn calc_mpr(prob: *const MfProblem, model: *const MfModel, tranpose: bool) -> c_double;
pub fn calc_auc(prob: *const MfProblem, model: *const MfModel, tranpose: bool) -> c_double;
pub fn calc_mpr(prob: *const MfProblem, model: *const MfModel, transpose: bool) -> c_double;
pub fn calc_auc(prob: *const MfProblem, model: *const MfModel, transpose: bool) -> c_double;
}

8
src/model.rs
View File

@@ -135,16 +135,16 @@ impl Model {
unsafe { calc_accuracy(&prob, self.model) }
}
pub fn mpr(&self, data: &Matrix, tranpose: bool) -> f64 {
pub fn mpr(&self, data: &Matrix, transpose: bool) -> f64 {
assert!(self.is_fit());
let prob = data.to_problem();
unsafe { calc_mpr(&prob, self.model, tranpose) }
unsafe { calc_mpr(&prob, self.model, transpose) }
}
pub fn auc(&self, data: &Matrix, tranpose: bool) -> f64 {
pub fn auc(&self, data: &Matrix, transpose: bool) -> f64 {
assert!(self.is_fit());
let prob = data.to_problem();
unsafe { calc_auc(&prob, self.model, tranpose) }
unsafe { calc_auc(&prob, self.model, transpose) }
}
fn with_model(model: *const MfModel) -> Self {