Merge pull request #57 from HyperCodec/examples-indicatif

Add indicatif progress bars for examples and increase generation count.

Author: HyperCodec

Cargo.lock

@@ -28,6 +28,7 @@ serde = ["dep:serde", "dep:serde-big-array"]
 [dependencies]
 bitflags = "2.5.0"
 genetic-rs = { version = "0.5.1", features = ["derive"] }
+
 lazy_static = "1.4.0"
 rand = "0.8.5"
 rayon = { version = "1.8.1", optional = true }
@@ -37,4 +38,5 @@ serde-big-array = { version = "0.5.1", optional = true }
 [dev-dependencies]
 bincode = "1.3.3"
 serde_json = "1.0.114"
-plotters = "0.3.5"
+plotters = "0.3.5"
+indicatif = "0.17.8"

Cargo.toml

@@ -1,5 +1,6 @@
 //! A basic example of NEAT with this crate. Enable the `crossover` feature for it to use crossover reproduction
 
+use indicatif::{ProgressBar, ProgressStyle};
 use neat::*;
 use rand::prelude::*;
 
@@ -103,10 +104,23 @@ fn main() {
         crossover_pruning_nextgen,
     );
 
-    for _ in 0..100 {
+    const GENS: u64 = 1000;
+    let pb = ProgressBar::new(GENS)
+        .with_style(
+            ProgressStyle::with_template(
+                "[{elapsed_precise}] {bar:40.cyan/blue} | {msg} {pos}/{len}",
+            )
+            .unwrap(),
+        )
+        .with_message("gen");
+
+    for _ in 0..GENS {
         sim.next_generation();
+        pb.inc(1);
     }
 
+    pb.finish();
+
     #[cfg(not(feature = "serde"))]
     let mut fits: Vec<_> = sim.genomes.iter().map(fitness).collect();
 

examples/basic.rs

@@ -3,6 +3,7 @@ use std::{
     sync::{Arc, Mutex},
 };
 
+use indicatif::{ProgressBar, ProgressStyle};
 use neat::*;
 use plotters::prelude::*;
 use rand::prelude::*;
@@ -73,7 +74,7 @@ struct PerformanceStats {
 }
 
 const OUTPUT_FILE_NAME: &'static str = "fitness-plot.svg";
-const GENS: usize = 100;
+const GENS: usize = 1000;
 
 fn main() -> Result<(), Box<dyn Error>> {
     #[cfg(not(feature = "rayon"))]
@@ -94,12 +95,25 @@ fn main() -> Result<(), Box<dyn Error>> {
         ng,
     );
 
+    let pb = ProgressBar::new(GENS as u64)
+        .with_style(
+            ProgressStyle::with_template(
+                "[{elapsed_precise}] {bar:40.cyan/blue} | {msg} {pos}/{len}",
+            )
+            .unwrap(),
+        )
+        .with_message("gen");
+
     println!("Training...");
 
     for _ in 0..GENS {
         sim.next_generation();
+
+        pb.inc(1);
     }
 
+    pb.finish();
+
     // prevent `Arc::into_inner` from failing
     drop(sim);
 
@@ -116,7 +130,7 @@ fn main() -> Result<(), Box<dyn Error>> {
         .margin(5)
         .x_label_area_size(30)
         .y_label_area_size(30)
-        .build_cartesian_2d(0usize..100, 0f32..200.0)?;
+        .build_cartesian_2d(0usize..GENS, 0f32..1000.0)?;
 
     chart.configure_mesh().draw()?;