马维尔

用于 Rust 编程语言的神经网络库。

马维尔实现了多层感知器神经网络和两种学习算法：反向传播和遗传算法。

更多文档请参阅文档。

lib.rs:

多层感知器、反向传播算法和一些遗传算法的实现。使用遗传算法

extern crate maver;
use maver::*;

struct XOR;

impl Environment for XOR {
   fn test(&self, org: &mut Organism) {
       let mut distance: f64;
       let nn = &mut org.nn;
       let out = nn.forward_prop(&[0.0, 0.0]);
       distance = (0.0 - out[0]).powi(2);
       let out = nn.forward_prop(&[0.0, 1.0]);
       distance += (1.0 - out[0]).powi(2);
       let out = nn.forward_prop(&[1.0, 0.0]);
       distance += (1.0 - out[0]).powi(2);
       let out = nn.forward_prop(&[1.0, 1.0]);
       distance += (0.0 - out[0]).powi(2);
       let fitness = 16.0 / (1.0 + distance);
       org.fitness = fitness;
   }
}

fn main() {
   let mut n = Genetic::new(100, &[2, 1], Activation::Tanh, Activation::Tanh);
   let p = LearnParams::default();
   let mut champion = None;
   let mut i = 0;
   while champion.is_none() {
       n.evaluate(&mut XOR);
       if n.get_champion().fitness > 15.9 {
           champion = Some(n.get_champion().clone());
       }
       println!(
           "Iteration: {:?},best fitness: {}",
          i,
           n.get_champion().fitness
       );
       n.evolve(&p);

      i += 1;
   }
   println!("{:#?}", champion.as_ref().unwrap());
   println!(
       "{:?}",
       champion.as_mut().unwrap().nn.forward_prop(&[0.0, 0.0])
   );
   println!(
       "{:?}",
       champion.as_mut().unwrap().nn.forward_prop(&[1.0, 0.0])
   );
   println!(
       "{:?}",
       champion.as_mut().unwrap().nn.forward_prop(&[1.0, 1.0])
   );
   println!(
       "{:?}",
       champion.as_mut().unwrap().nn.forward_prop(&[0.0, 1.0])
   );
}

使用反向传播算法

let mut nn = NeuralNetwork!(vec![2,2,1],Activation::Tanh,Activation::Tanh);

let xor_sets = vec![
   (vec![0., 0.], vec![0.0]),
   (vec![0., 1.], vec![1.0]),
   (vec![1., 0.], vec![1.0]),
   (vec![1., 1.], vec![0.0]),
];

nn.train(1000,false,xor_sets,Some(0.01));
println!("{:?}",nn.forward_prop(&[0.0,0.0]));
println!("{:?}",nn.forward_prop(&[0.0,1.0]));