astrai 2.2.0

A pretty bad neural network 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
24
25
26
27
28
29

use super::*;

pub type NeuronID = (usize, usize); // layer, index

impl Network {
	// TODO: move to initialization context
    pub fn add_connection(&mut self, in_neuron_id: NeuronID, out_neuron_id: NeuronID, weight: f64) {
        self.connections.push(Connection {
            in_neuron_id,
            out_neuron_id,
            weight,
            enabled: true,
        });
    }

    pub fn get_input_layer(&self) -> Layer {
        self.layers.last().unwrap().clone()
    }

    pub fn get_output_layer(&self) -> Layer {
        self.layers.last().unwrap().clone()
    }

    pub fn zero_out_network(&mut self) {
        for layer in self.layers.iter_mut() {
            layer.zero_out();
        }
	}
}