entrenar/train/metrics/mod.rs
1//! Evaluation metrics for training and validation
2//!
3//! This module provides common metrics for evaluating model performance:
4//! - Classification: Accuracy, Precision, Recall, F1
5//! - Regression: MSE, MAE, R²
6
7mod classification;
8mod regression;
9mod trait_def;
10
11#[cfg(test)]
12mod tests;
13
14// Re-export the Metric trait
15pub use trait_def::Metric;
16
17// Re-export classification metrics
18pub use classification::{Accuracy, F1Score, Precision, Recall};
19
20// Re-export regression metrics
21pub use regression::{R2Score, MAE, RMSE};