fix: metric parameter name changed

src/metrics/mod.rs

@@ -21,16 +21,32 @@ impl ClassificationMetrics {
     pub fn precision() -> precision::Precision {
         precision::Precision {}
     }
+
+    pub fn f1() -> f1::F1 {
+        f1::F1 {}
+    }
+
+    pub fn roc_auc_score() -> auc::AUC {
+        auc::AUC {}
+    }
 }
 
-pub fn accuracy<T: FloatExt, V: BaseVector<T>>(y_true: &V, y_prod: &V) -> T {
-    ClassificationMetrics::accuracy().get_score(y_true, y_prod)
+pub fn accuracy<T: FloatExt, V: BaseVector<T>>(y_true: &V, y_pred: &V) -> T {
+    ClassificationMetrics::accuracy().get_score(y_true, y_pred)
 }
 
-pub fn recall<T: FloatExt, V: BaseVector<T>>(y_true: &V, y_prod: &V) -> T {
-    ClassificationMetrics::recall().get_score(y_true, y_prod)
+pub fn recall<T: FloatExt, V: BaseVector<T>>(y_true: &V, y_pred: &V) -> T {
+    ClassificationMetrics::recall().get_score(y_true, y_pred)
 }
 
-pub fn precision<T: FloatExt, V: BaseVector<T>>(y_true: &V, y_prod: &V) -> T {
-    ClassificationMetrics::precision().get_score(y_true, y_prod)
+pub fn precision<T: FloatExt, V: BaseVector<T>>(y_true: &V, y_pred: &V) -> T {
+    ClassificationMetrics::precision().get_score(y_true, y_pred)
+}
+
+pub fn f1<T: FloatExt, V: BaseVector<T>>(y_true: &V, y_pred: &V) -> T {
+    ClassificationMetrics::f1().get_score(y_true, y_pred)
+}
+
+pub fn roc_auc_score<T: FloatExt, V: BaseVector<T>>(y_true: &V, y_pred_probabilities: &V) -> T {
+    ClassificationMetrics::roc_auc_score().get_score(y_true, y_pred_probabilities)
 }