Changeset 1128 for framspy

Timestamp:

04/11/21 03:21:14 (4 years ago)

Author:

Maciej Komosinski

Message:

Introduced the number of generations as a command-line parameter; refactored; fixed a typo

Location:

framspy/evolalg

Files:

• examples/invalid.py (modified) (2 diffs)
• examples/niching_novelty.py (modified) (9 diffs)
• examples/standard.py (modified) (2 diffs)
• statistics/halloffame_stats.py (modified) (1 diff)

framspy/evolalg/examples/invalid.py

@@ -94 +94 @@
                    statistics_union]
 
-    end_steps = [PopulationSave("halloffame.gen", provider=hall_of_fame.haloffame, fields={"genotype": "genotype",
+    end_steps = [PopulationSave("halloffame.gen", provider=hall_of_fame.halloffame, fields={"genotype": "genotype",
                                                                                            "fitness": "fitness",
                                                                                            "custom": "recording"})]
@@ -107 +107 @@
     experiment.init()
     experiment.run(3)
-    for ind in hall_of_fame.haloffame:
+    for ind in hall_of_fame.halloffame:
         print("%g\t%s" % (ind.fitness, ind.genotype))
 

framspy/evolalg/examples/niching_novelty.py

@@ -1 +1 @@
 import argparse
 import os
+import pickle
 import sys
 from enum import Enum
@@ -8 +9 @@
 from FramsticksLib import FramsticksLib
 from evolalg.base.lambda_step import LambdaStep
+from evolalg.base.step import Step
 from evolalg.dissimilarity.frams_dissimilarity import FramsDissimilarity
 from evolalg.dissimilarity.levenshtein import LevenshteinDissimilarity
@@ -21 +23 @@
 from evolalg.base.union_step import UnionStep
 from evolalg.utils.population_save import PopulationSave
+import time
 
 
@@ -51 +54 @@
         description='Run this program with "python -u %s" if you want to disable buffering of its output.' % sys.argv[
             0])
-    parser.add_argument('-path', type=ensureDir, required=True, help='Path to Framsticks without trailing slash.')
+    parser.add_argument('-path', type=ensureDir, required=True, help='Path to the Framsticks library without trailing slash.')
     parser.add_argument('-opt', required=True,
                         help='optimization criteria : vertpos, velocity, distance, vertvel, lifespan, numjoints, numparts, numneurons, numconnections. Single or multiple criteria.')
@@ -63 +66 @@
                         help=' Fitness criteria DEFAULT = raw', choices=list(Fitness))
     parser.add_argument('-popsize', type=int, default=50, help="Size of population, default 50.")
+    parser.add_argument('-generations', type=int, default=5, help="Number of generations, default 5.")
     parser.add_argument('-num_parts', type=int, default=None, help="Maximum number of parts. Default None")
-    parser.add_argument('-checkpoint_path', required=False, default=None, help="Path to checkpoint path")
+    parser.add_argument('-checkpoint_path', required=False, default=None, help="Path to the checkpoint file")
     parser.add_argument('-checkpoint_interval', required=False, type=int, default=100, help="Checkpoint interval")
     return parser.parse_args()
@@ -96 +100 @@
 
 
-def main():
-    print("Running experiment with", sys.argv)
+def load_experiment(path):
+    with open(path, "rb") as file:
+        experiment = pickle.load(file)
+    print("Loaded experiment. Generation:", experiment.generation)
+    return experiment
+
+
+def create_experiment():
     parsed_args = parseArguments()
     frams = FramsticksLib(parsed_args.path, parsed_args.lib,
                           parsed_args.sim)
     # Steps for generating first population
-    init_stages = [FramsPopulation(frams, parsed_args.genformat, parsed_args.popsize)]
+    init_stages = [
+        FramsPopulation(frams, parsed_args.genformat, parsed_args.popsize)
+    ]
 
     # Selection procedure
-    selection = TournamentSelection(5, copy=True) # 'fitness' by default, the targeted attribute can be changed, e.g. fit_attr="fitness_raw"
+    selection = TournamentSelection(5,
+                                    copy=True)  # 'fitness' by default, the targeted attribute can be changed, e.g. fit_attr="fitness_raw"
 
     # Procedure for generating new population. This steps will be run as long there is less than
@@ -193 +206 @@
     end_stages = [
         fitness_end,
-        PopulationSave("halloffame.gen", provider=hall_of_fame.haloffame, fields={"genotype": "genotype",
-                                                                                            "fitness": "fitness_raw"})]
+        PopulationSave("halloffame.gen", provider=hall_of_fame.halloffame, fields={"genotype": "genotype",
+                                                                                  "fitness": "fitness_raw"})]
     # ...but custom fields can be added, e.g. "custom": "recording"
 
@@ -209 +222 @@
                             checkpoint_interval=parsed_args.checkpoint_interval
                             )
-    experiment.init()  # init is mandatory
-    experiment.run(10)
+    return experiment
+
+
+def main():
+    print("Running experiment with", sys.argv)
+    parsed_args = parseArguments()
+
+    if os.path.exists(parsed_args.checkpoint_path):
+        experiment = load_experiment(parsed_args.checkpoint_path)
+        FramsticksLib(parsed_args.path, parsed_args.lib,
+                      parsed_args.sim)
+    else:
+        experiment = create_experiment()
+        experiment.init()  # init is mandatory
+
+
+    experiment.run(parsed_args.generations)
+
     # Next call for experiment.run(10) will do nothing. Parameter 10 specifies how many generations should be
     # in one experiment. Previous call generated 10 generations.
@@ -226 +255 @@
     # # All work produced by first run will be "destroyed" by second init().
 
-    for ind in hall_of_fame.haloffame:
-        print("%g\t%s" % (ind.fitness, ind.genotype))
 
 

framspy/evolalg/examples/standard.py

@@ -42 +42 @@
 
     parser.add_argument("-popsize", type=int, default=50, help="Size of population, default 50.")
+    parser.add_argument('-generations', type=int, default=5, help="Number of generations, default 5.")
     return parser.parse_args()
 
@@ -96 +97 @@
                             )
     experiment.init()
-    experiment.run(3)
-    for ind in hall_of_fame.haloffame:
+    experiment.run(parsed_args.generations)
+    for ind in hall_of_fame.halloffame:
         print("%g\t%s" % (ind.fitness, ind.genotype))
 

framspy/evolalg/statistics/halloffame_stats.py

@@ -5 +5 @@
 class HallOfFameStatistics(Statistics):
     def __init__(self, size, fields="fitness"):
-        self.haloffame = HallOfFameCustom(size, fitness_field=fields)
+        self.halloffame = HallOfFameCustom(size, fitness_field=fields)
 
     def init(self):
-        self.haloffame.clear()
+        self.halloffame.clear()
 
     def collect(self, population):
-        self.haloffame.update(population)
+        self.halloffame.update(population)