simple_nn

Simple neural network implementation in Rust.

NOTE: I wanted to give Rust a try, and decided to try implementing a simple NN framework, but this is not meant to be used in production (the current implementation is way too slow for now anyway).

Here is a small example for the mnist dataset.

extern crate simple_nn;

use simple_nn::{nn, utils};

fn main() {
    let mut network = nn::NetworkBuilder::new()
        .add(nn::layers::Dense::new(784, 100))
        .add(nn::layers::Relu::new())
        .add(nn::layers::Dense::new(100, 100))
        .add(nn::layers::Relu::new())
        .add(nn::layers::Dense::new(100, 10))
        .add_output(nn::layers::Softmax::new())
        .minimize(nn::objectives::CrossEntropy::new())
        .with(nn::optimizers::SGD::new(0.5))
        .build();

    println!("loading training data...");

    let x_train = utils::loader::matrix_from_txt("data/train_x_60000x784_float32.txt").unwrap().transform(|v: f64| v / 255.0);
    let y_train = utils::loader::matrix_from_txt("data/train_y_60000_int32.txt").unwrap().to_one_hot(10);

    let train_options = nn::TrainOptions::default().with_epochs(5).with_batch_size(256);
    network.fit(&x_train, &y_train, train_options);

    println!("loading test data...");

    let x_test = utils::loader::matrix_from_txt("data/test_x_10000x784_float32.txt").unwrap().transform(|v: f64| v / 255.0);
    let y_test = utils::loader::matrix_from_txt("data/test_y_10000_int32.txt").unwrap().to_one_hot(10);

    let predict_probs = network.predict_probs(&x_test);
    let loss = network.mean_loss_from_probs(&predict_probs, &y_test);
    let accuracy = network.accuracy_from_probs(&predict_probs, &y_test);
    println!("accuracy = {}, mean loss = {}", accuracy, loss);
}

Progress

Only very few functions have been implemented yet. Help is very welcome.

Layers

• Dense (missing bias)
• Dropout
• Convolutional

Activations

• ReLU
• sigmoid
• tanh
• softplus
• softsign

Objectives

• Categorical Cross Entropy
• Binary Cross Entropy
• Mean square
• Poisson
• KL divergence

Optimizers

• SGD
• Adam
• Adamax
• RMSprop

Other

• Serialization
• Metrics
• Layer configurations