3 个版本

0.1.2 2024 年 2 月 9 日
0.1.1 2023 年 11 月 24 日
0.1.0 2023 年 4 月 12 日

#224机器学习


用于 zbusdg

MIT/Apache 许可协议

25KB
381 行(不含注释)

tests Crates.io docs.rs

关于 rhai-ml

此crate为Rhai脚本语言提供了一些基本的机器学习和人工智能工具。要获取完整的API参考,请查看文档

安装

要使用rhai-ml的最新发布版本,将以下内容添加到您的Cargo.toml文件中:

rhai-ml = "0.1.2"

要使用最新版本,请添加以下内容:

rhai-ml = { git = "https://github.com/cmccomb/rhai-ml" }

使用方法

使用此crate非常简单!如果您只想评估一行Rhai,那么您只需要:

use rhai::FLOAT;
use rhai_ml::eval;
let result = eval::<FLOAT>("\
let xdata = [[1.0, 2.0], [2.0, 3.0], [3.0, 4.0]]; \
let ydata = [1.0, 2.0, 3.0]; \
let model = train(xdata, ydata, \"linear\"); \
let ypred = predict(xdata, model);
ypred[0]
").unwrap();

如果您需要将rhai-ml作为持久Rhai脚本引擎的一部分使用,则可以这样做:

use rhai::{Engine, packages::Package, FLOAT};
use rhai_ml::MLPackage;

// Create a new Rhai engine
let mut engine = Engine::new();

// Add the rhai-ml package to the new engine
engine.register_global_module(MLPackage::new().as_shared_module());

// Now run your code
let value = engine.eval::<FLOAT>("\
let xdata = [[1.0, 2.0], [2.0, 3.0], [3.0, 4.0]]; \
let ydata = [1.0, 2.0, 3.0]; \
let model = train(xdata, ydata, \"linear\"); \
let ypred = predict(xdata, model);
ypred[0]
").unwrap();

特性

特性 默认 描述
metadata 禁用 启用导出函数元数据,这对于在Rhai示例上运行doc-tests是必要的。

依赖项

~7MB
~139K SLoC