Crate sklearn_rs 

sklearn-rs

一个受 scikit-learn 启发的 Rust 机器学习库。

设计理念

• 一致的 API 设计
• 强类型安全
• 高性能计算
• 完善的错误处理

快速开始

use sklearn_rs::{LinearRegression, Estimator, Predictor};
use ndarray::array;

// 创建示例数据
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);

Re-exports

pub use base::Estimator;

pub use base::Predictor;

pub use base::Transformer;

pub use error::SklearnError;

pub use error::Result;

pub use linear_model::LinearRegression;

pub use linear_model::LinearModel;

Modules

base

error

linear_model

metrics