9个不稳定版本 (3个破坏性更新)
| 0.4.1 | 2022年10月1日 |
|---|---|
| 0.4.0 | 2022年9月29日 |
| 0.3.0 | 2022年9月22日 |
| 0.2.4 | 2022年7月30日 |
| 0.1.0 | 2022年7月23日 |
#719 in 机器学习
每月39次下载
用于 entity-gym-rs
165KB
2K SLoC
RogueNet Rust
rogue-net crate提供了RogueNet神经网络的纯Rust实现。它可以用来加载使用Entity Neural Network Trainer创建的代理，并在Rust应用程序中使用。
use std::collections::HashMap;
use ndarray::prelude::*;
use rogue_net::RogueNet;
let rogue_net = RogueNet::load("checkpoint-dir");
let mut entities = HashMap::new();
entities.insert("Head".to_string(), array![[3.0, 4.0]]);
entities.insert("SnakeSegment".to_string(), array![[3.0, 4.0], [4.0, 4.0]]);
entities.insert("Food".to_string(), array![[3.0, 5.0], [8.0, 4.0]]);
let (action_probs, actions) = rogue_net.forward(&entities);
依赖项
~14–24MB
~376K SLoC