你喜欢这个仓库并想支持它吗？如果是这样，留下一个 ⭐

如何使用

在处理此 crate 时，您需要在代理的 DNA 中使用 NeuralNetworkTopology 结构，并在最终想要测试其性能时使用 NeuralNetwork::from。此 crate 的其余部分也随此 crate 一起重新导出。

以下是一个使用此 crate 的示例

use neat::*;

#[derive(Clone, RandomlyMutable, DivisionReproduction)]
struct MyAgentDNA {
    network: NeuralNetworkTopology<1, 2>,
}

impl GenerateRandom for MyAgentDNA {
    fn gen_random(rng: &mut impl rand::Rng) -> Self {
        Self {
            network: NeuralNetworkTopology::new(0.01, 3, rng),
        }
    }
}

struct MyAgent {
    network: NeuralNetwork<1, 2>,
    // ... other state
}

impl From<&MyAgentDNA> for MyAgent {
    fn from(value: &MyAgentDNA) -> Self {
        Self {
            network: NeuralNetwork::from(&value.network),
        }
    }
}

fn fitness(dna: &MyAgentDNA) -> f32 {
    // agent will simply try to predict whether a number is greater than 0.5
    let mut agent = MyAgent::from(dna);
    let mut rng = rand::thread_rng();
    let mut fitness = 0;

    // use repeated tests to avoid situational bias and some local maximums, overall providing more accurate score
    for _ in 0..10 {
        let n = rng.gen::<f32>();
        let above = n > 0.5;

        let res = agent.network.predict([n]);
        let resi = res.iter().max_index();

        if resi == 0 ^ above {
            // agent did not guess correctly, punish slightly (too much will hinder exploration)
            fitness -= 0.5;

            continue;
        }

        // agent guessed correctly, they become more fit.
        fitness += 3.;
    }

    fitness
}

fn main() {
    let mut rng = rand::thread_rng();

    let mut sim = GeneticSim::new(
        Vec::gen_random(&mut rng, 100),
        fitness,
        division_pruning_nextgen,
    );

    // simulate 100 generations
    for _ in 0..100 {
        sim.next_generation();
    }

    // display fitness results
    let fits: Vec<_> = sim.entities
        .iter()
        .map(fitness)
        .collect();

    dbg!(&fits, fits.iter().max());
}

许可证

此 crate 适用于 MIT 许可证