scholar 0.1.7

A supervised machine learning library.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

use scholar::{Dataset, NeuralNet, Sigmoid};

fn main() -> anyhow::Result<()> {
    let data = vec![
        (vec![0.0, 0.0], vec![0.0]),
        (vec![0.0, 1.0], vec![0.0]),
        (vec![1.0, 0.0], vec![0.0]),
        (vec![1.0, 1.0], vec![1.0]),
    ];

    let dataset = Dataset::from(data);

    let mut brain: NeuralNet<Sigmoid> = NeuralNet::new(&[2, 2, 1]);
    brain.train(dataset, 250_000, 0.01);

    brain.save("examples/brain.network")?;

    let mut brain: NeuralNet<Sigmoid> = NeuralNet::from_file("examples/brain.network")?;

    println!("Prediction: {:.2}", brain.guess(&[1.0, 1.0])[0]);

    Ok(())
}