rusty-ai

包含各种机器学习算法的 Rust 库

lib.rs:

Rusty-ai

rusty-ai 使用 Rust 实现了各种分类和回归算法。它还包含一些数据操作和度量工具。

入门指南

要使用 rusty-ai，请在您的 Cargo.toml 文件中添加以下内容

[dependencies]
rusty-ai = "*"

示例用法

以下是如何使用 rusty-ai 在示例数据集上训练高斯朴素贝叶斯分类器的快速示例

use rusty_ai::bayes::gaussian::*;
use rusty_ai::data::dataset::*;
use nalgebra::{DMatrix, DVector};

let x = DMatrix::from_row_slice(4, 2, &[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0]);
let y = DVector::from_vec(vec![0, 0, 1, 1]);

let dataset = Dataset::new(x, y);

let mut model = GaussianNB::new();

model.fit(&dataset).unwrap();

let test_x = DMatrix::from_row_slice(2, 2, &[1.0, 2.0, 3.0, 4.0]);

let predictions = model.predict(&test_x).unwrap();