2个版本
0.1.1 | 2022年3月4日 |
---|---|
0.1.0 | 2022年3月4日 |
#492 in 机器学习
7KB
87 行
syron
syron是一个旨在轻松构建简单神经网络的Rust库。
示例
use syron::nn::{Network, Activation};
fn main() {
// XOR Problem
let x = ndarray::arr2(&[[0.0, 0.0], [0.0, 1.0], [1.0, 0.0], [1.0, 1.0]]);
let y = ndarray::arr2(&[[0.0], [1.0], [1.0], [0.0]]);
// Create a network with 2 inputs and 3 hidden layers
let mut nn = Network::new(2, 3, Activation::Sigmoid);
nn.train(x, y, 1000);
println!("{:?}", nn.predict(ndarray::arr2(&[[0.0, 0.0]])));
}
lib.rs
:
syron是一个旨在轻松构建简单神经网络的Rust库。
示例
use syron::nn::{Network, Activation};
fn main() {
// XOR Problem
let x = ndarray::arr2(&[[0.0, 0.0], [0.0, 1.0], [1.0, 0.0], [1.0, 1.0]]);
let y = ndarray::arr2(&[[0.0], [1.0], [1.0], [0.0]]);
// Create a network with 2 inputs and 3 hidden layers
let mut nn = Network::new(2, 3, Activation::Sigmoid);
nn.train(x, y, 1000);
println!("{:?}", nn.predict(ndarray::arr2(&[[0.0, 0.0]])));
}
依赖项
~2.5MB
~44K SLoC