#neural-network #scientific #machine-learning #development-tools-debuggings

syron

Syron是一个用于构建、训练和运行基本神经网络的Rust库

2个版本

0.1.1 2022年3月4日
0.1.0 2022年3月4日

#492 in 机器学习

MIT/Apache

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