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(())
}