Add CustomReporter for examples/evolve_scrabble to showcase genes_key_cardinality v fitness_score_cardinality #7

Author: basvanwesting

examples/evolve_scrabble.rs

@@ -4,14 +4,14 @@ use std::collections::{HashMap, HashSet};
 type Row = usize;
 type Column = usize;
 
-#[derive(Clone, Copy, Debug, Eq, PartialEq)]
-enum Orientation {
+#[derive(Clone, Copy, Debug, Eq, PartialEq, Hash)]
+pub enum Orientation {
     Horizontal,
     Vertical,
 }
 
-#[derive(Clone, Copy, Debug, Eq, PartialEq)]
-struct WordPosition(pub Row, pub Column, pub Orientation);
+#[derive(Clone, Copy, Debug, Eq, PartialEq, Hash)]
+pub struct WordPosition(pub Row, pub Column, pub Orientation);
 impl Allele for WordPosition {}
 
 #[derive(Clone, Debug)]
@@ -223,6 +223,65 @@ impl ScrabbleFitness {
     }
 }
 
+use cardinality_estimator::CardinalityEstimator;
+
+#[derive(Clone)]
+pub struct CustomReporter(usize);
+impl StrategyReporter for CustomReporter {
+    type Genotype = MultiListGenotype<WordPosition>;
+
+    fn on_enter<S: StrategyState<Self::Genotype>, C: StrategyConfig>(
+        &mut self,
+        genotype: &Self::Genotype,
+        state: &S,
+        config: &C,
+    ) {
+        let number_of_seed_genes = genotype.seed_genes_list().len();
+        if number_of_seed_genes > 0 {
+            println!(
+                "enter - {}, iteration: {}, number of seed genes: {}",
+                config.variant(),
+                state.current_iteration(),
+                number_of_seed_genes
+            );
+        } else {
+            println!(
+                "enter - {}, iteration: {}",
+                config.variant(),
+                state.current_iteration()
+            );
+        }
+    }
+
+    fn on_new_generation<S: StrategyState<Self::Genotype>, C: StrategyConfig>(
+        &mut self,
+        _genotype: &Self::Genotype,
+        state: &S,
+        _config: &C,
+    ) {
+        if state.current_generation() % self.0 == 0 {
+            let mut estimator = CardinalityEstimator::<u64>::new();
+            state
+                .population_as_ref()
+                .chromosomes
+                .iter()
+                .map(|c| c.genes_key())
+                .for_each(|key| estimator.insert(&key));
+            let genes_key_cardinality = estimator.estimate();
+            let fitness_score_cardinality = state.population_as_ref().fitness_score_cardinality();
+
+            println!(
+                "current_generation: {}, stale_generations: {}, best_generation: {}, fitness_score_cardinality: {}, genes_key_cardinality: {}",
+                state.current_generation(),
+                state.stale_generations(),
+                state.best_generation(),
+                fitness_score_cardinality,
+                genes_key_cardinality,
+            );
+        }
+    }
+}
+
 fn main() {
     env_logger::init();
 
@@ -283,7 +342,8 @@ fn main() {
         // .with_reporter(EvolveReporterSimple::new_with_flags(
         //     100, false, false, false, true,
         // ))
-        .with_reporter(EvolveReporterSimple::new(100))
+        // .with_reporter(EvolveReporterSimple::new(100))
+        .with_reporter(CustomReporter(50))
         .with_par_fitness(true)
         .with_fitness(ScrabbleFitness::new(
             words.clone(),