vexus 4.0.0

A neural network builder and trainer struct to make your own AI models.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

use vexus::{NeuralNetwork, Sigmoid};

fn main() {
    // Create a neural network with 2 inputs, one hidden layer of 4 neurons, and 1 output
    let mut nn = NeuralNetwork::new(vec![2, 4, 1], 0.1, Box::new(Sigmoid));

    // Training data for XOR
    let training_data = vec![
        (vec![0.0, 0.0], vec![0.0]),
        (vec![0.0, 1.0], vec![1.0]),
        (vec![1.0, 0.0], vec![1.0]),
        (vec![1.0, 1.0], vec![0.0]),
    ];

    // Train the network
    for _ in 0..1000000 {
        for (inputs, expected) in &training_data {
            let _outputs = nn.forward(inputs);
            nn.backpropagate(expected);
        }
    }

    // Test the network
    for (inputs, expected) in &training_data {
        let outputs = nn.forward(inputs);
        println!(
            "Input: {:?}, Expected: {:?}, Got: {:.4}",
            inputs, expected[0], outputs[0]
        );
    }
}