neural_networks

使用 Rust 编写的神经网络库。

MNIST 示例

步骤 1：在 Cargo.toml 的依赖项中添加 neural_networks="0.0.1"。

步骤 2：从项目的 GitHub 仓库的数据集文件夹下载 mnist 数据集。解压它。

步骤 3：现在您可以使用以下 mnist 示例中的 neural_networks 函数。

use neural_networks::neural_networks::NeuralNetwork;
use neural_networks::{get_accuracy, train};

fn main() {
    let mut nn = NeuralNetwork::new(784, vec![8, 8], 10, 0.1, "sigmoid");
    train(&mut nn, "mnist_train.csv", 5);
    let acc = get_accuracy(&nn, "mnist_test.csv") * 100.0;
    println!("Accuarcy: {}%", acc);
}