rustorch 0.6.29

//! 学習ループの抽象化
//! Training loop abstraction
//!
//! このモジュールは機械学習の訓練プロセスを抽象化し、
//! 再利用可能で拡張可能な学習ループを提供します。
//!
//! ## 主要コンポーネント
//!
//! - \[`Trainer`\]: 汎用的な学習ループの実装
//! - \[`TrainingState`\]: 訓練状態の管理
//! - \[`Metrics`\]: メトリクスの収集と計算
//! - \[`EarlyStopping`\]: 早期停止の実装
//! - \[`Checkpoint`\]: モデルの保存と復元

pub mod callbacks;
pub mod checkpoint;
pub mod metrics;
pub mod state;
pub mod trainer;

// #[cfg(test)]
// pub mod tests;  // 一時的にコメントアウト - SGDインポート問題のため

#[cfg(test)]
pub mod simple_tests;

pub use callbacks::{Callback, EarlyStopping, LearningRateScheduler, ProgressBar};
pub use checkpoint::{CheckpointManager, SaveConfig};
pub use metrics::{MetricsCollector, TrainingMetrics};
pub use state::{BatchState, EpochState, TrainingState};
pub use trainer::{TrainableModel, Trainer, TrainerBuilder, TrainerConfig};