sklearn_rs/
lib.rs

1//! # sklearn-rs
2//!
3//! 一个受 scikit-learn 启发的 Rust 机器学习库。
4//!
5//! ## 设计理念
6//!
7//! - 一致的 API 设计
8//! - 强类型安全
9//! - 高性能计算
10//! - 完善的错误处理
11//!
12//! ## 快速开始
13//!
14//! ```rust
15//! use sklearn_rs::{LinearRegression, Estimator, Predictor};
16//! use ndarray::array;
17//!
18//! # fn main() -> Result<(), Box<dyn std::error::Error>> {
19//! // 创建示例数据
20//! let x = array![
21//!     [1.0, 1.0],
22//!     [1.0, 2.0],
23//!     [2.0, 2.0],
24//!     [2.0, 3.0]
25//! ];
26//! let y = array![1.0, 1.0, 2.0, 2.0];
27//!
28//! // 创建和训练模型
29//! let regressor = LinearRegression::default();
30//! let model = regressor.fit(&x, &y)?;
31//!
32//! // 进行预测
33//! let predictions = model.predict(&x)?;
34//! println!("预测结果: {:?}", predictions);
35//! # Ok(())
36//! # }
37//! ```
38
39pub mod error;
40pub mod base;
41pub mod linear_model;
42pub mod metrics;
43
44// 重新导出主要类型和 trait
45pub use base::{Estimator, Predictor, Transformer};
46pub use error::{SklearnError, Result};
47pub use linear_model::{LinearRegression, LinearModel};