3个版本
新版本 0.1.2 | 2024年8月16日 |
---|---|
0.1.1 | 2024年8月16日 |
0.1.0 | 2024年8月16日 |
#4 in #neuron
每月下载量 152
55KB
1.5K SLoC
CTRNN
Rust的CTRNN框架。
网络类型
HashNetwork
: 具有灵活HashMap神经元连接的网络,非常适合大型稀疏网络。SsmNetwork
: 使用集中式突触强度矩阵的单线程网络。PowerNetwork
: 使用集中式突触强度矩阵的多线程、可分配网络。
用法
use ctrnn::PowerNetwork;
// PowerNetwork(size, d_in, d_out, worker_cores)
let mut power = PowerNetwork::new(24, 1, 1, 8).unwrap();
// PowerNetwork.weave(neural_density)
power.weave(0.8)
// PowerNetwork.forward(inputs, next_tau, step_size)
let inputs = vec![0.3];
let next_tau = ctrnn::get_ts() + 3.;
let step_size = 0.001;
power.forward(inputs, next_tau, step_size);
let losses: Vec<f64> = my_loss_fn();
let learning_rate = 0.01;
power.backward(learning_rate, losses).unwrap();
依赖关系
~3–6MB
~107K SLoC