sklearn-rs 0.1.0

A scikit-learn inspired machine learning library in Rust
Documentation
//! # sklearn-rs
//!
//! 一个受 scikit-learn 启发的 Rust 机器学习库。
//!
//! ## 设计理念
//!
//! - 一致的 API 设计
//! - 强类型安全
//! - 高性能计算
//! - 完善的错误处理
//!
//! ## 快速开始
//!
//! ```rust
//! use sklearn_rs::{LinearRegression, Estimator, Predictor};
//! use ndarray::array;
//!
//! # fn main() -> Result<(), Box<dyn std::error::Error>> {
//! // 创建示例数据
//! let x = array![
//!     [1.0, 1.0],
//!     [1.0, 2.0],
//!     [2.0, 2.0],
//!     [2.0, 3.0]
//! ];
//! let y = array![1.0, 1.0, 2.0, 2.0];
//!
//! // 创建和训练模型
//! let regressor = LinearRegression::default();
//! let model = regressor.fit(&x, &y)?;
//!
//! // 进行预测
//! let predictions = model.predict(&x)?;
//! println!("预测结果: {:?}", predictions);
//! # Ok(())
//! # }
//! ```

pub mod error;
pub mod base;
pub mod linear_model;
pub mod metrics;

// 重新导出主要类型和 trait
pub use base::{Estimator, Predictor, Transformer};
pub use error::{SklearnError, Result};
pub use linear_model::{LinearRegression, LinearModel};