use scholar::{Dataset, NeuralNet, Sigmoid};
fn main() {
let 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]),
];
let dataset = Dataset::from(data);
let mut brain: NeuralNet<Sigmoid> = NeuralNet::new(&[2, 10, 10, 1]);
brain.train(dataset, 250_000, 0.01);
println!("Prediction: {:.2}", brain.guess(&[1.0, 1.0])[0]);
}