1 个不稳定版本

0.1.0 2020年11月20日

#96 in #ml

MIT 许可证

23KB
574 代码行

mlflow-rs

Rust 库,提供对 MLflow REST API 的访问。

这不是一个官方项目。

示例

fn main() {
    let client = Client::for_server("http://127.0.0.1:5000/api");
    let experiment = client.create_experiment("My Experiment");

    for i in 0..3 {
        println!("Executing run {}", i);
        let run = experiment.create_run();

        run.log_param("i", &format!("{}", i));
        run.log_param("constant", "42");

        let mut rng = WyRand::new_seed(i.into());
        for s in 0..10 {
            let int: f64 = rng.generate::<u16>().into();
            let max: f64 = std::u16::MAX.into();
            let value = int / max;
            run.log_metric("rand", value, timestamp(), s);
        }
        run.terminate();
    }
}

状态

以下 API 部分已实现

  • 实验
    • 创建
    • 读取
    • 管理
  • 运行
    • 创建
    • 读取
    • 管理
  • 日志记录
    • 参数
    • 度量
    • 工件
  • 模型
    • 创建
    • 读取
    • 管理
  • 标签

依赖关系

~12MB
~335K SLoC