StackGP 0.0.5__tar.gz → 0.0.7__tar.gz

{stackgp-0.0.5 → stackgp-0.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: StackGP
-Version: 0.0.5
+Version: 0.0.7
 Summary: A stack-based genetic programming system
 Author-email: Nathan Haut <hautnath@msu.edu>
 Project-URL: Homepage, https://github.com/hoolagans/StackGP

{stackgp-0.0.5 → stackgp-0.0.7}/StackGP/StackGP.py

@@ -20,6 +20,7 @@ from sklearn.cluster import KMeans #for clustering in ensemble definition
 from scipy.optimize import minimize #for uncertainty maximization
 from sympy import symbols, simplify, expand
 import sympy as sym
+from IPython.display import display, clear_output
 
 import signal #for timing out functions
 from contextlib import contextmanager #for timing out functions
@@ -615,7 +616,7 @@ def alignGPModel(model, data, response): #Aligns a model
     setModelQuality(newModel,data,response)
     return newModel
 alignGPModel.__doc__ = "alignGPModel(model, input, response) aligns a model such that response-a*f(x)+b are minimized over a and b"
-def evolve(inputData, responseData, generations=100, ops=defaultOps(), const=defaultConst(), variableNames=[], mutationRate=79, crossoverRate=11, spawnRate=10, extinction=False,extinctionRate=10,elitismRate=50,popSize=300,maxComplexity=100,align=True,initialPop=[],timeLimit=300,capTime=False,tourneySize=5,tracking=False,modelEvaluationMetrics=[fitness,stackGPModelComplexity],dataSubsample=False,samplingMethod=randomSubsample,alternateObjectives=[],alternateObjFrequency=10):
+def evolve(inputData, responseData, generations=100, ops=defaultOps(), const=defaultConst(), variableNames=[], mutationRate=79, crossoverRate=11, spawnRate=10, extinction=False,extinctionRate=10,elitismRate=50,popSize=300,maxComplexity=100,align=True,initialPop=[],timeLimit=300,capTime=False,tourneySize=5,tracking=False,liveTracking=False,liveTrackingInterval=1,modelEvaluationMetrics=[fitness,stackGPModelComplexity],dataSubsample=False,samplingMethod=randomSubsample,alternateObjectives=[],alternateObjFrequency=10,allowEarlyTermination=False,earlyTerminationThreshold=0):
     
     metrics=modelEvaluationMetrics
 
@@ -627,6 +628,9 @@ def evolve(inputData, responseData, generations=100, ops=defaultOps(), const=def
     models=models+initialPop
     startTime=time.perf_counter()
     bestFits=[]
+    if liveTracking:
+        fig, ax = plt.subplots(figsize=(20,10))
+        ckTime=time.perf_counter()
     for i in range(generations):
         if capTime and time.perf_counter()-startTime>timeLimit:
             break
@@ -639,8 +643,22 @@ def evolve(inputData, responseData, generations=100, ops=defaultOps(), const=def
         for mods in models:
             setModelQuality(mods,inData,resData,modelEvaluationMetrics=metrics)
         models=removeIndeterminateModels(models)
-        if tracking:
+        if allowEarlyTermination and min([mods[2][0] for mods in models])<=earlyTerminationThreshold:
+            print("Early termination at generation ", i)
+            break
+        if tracking or liveTracking:
             bestFits.append(min([mods[2][0] for mods in paretoTournament(models)]))
+        if liveTracking and time.perf_counter()-ckTime>liveTrackingInterval:
+            ax.clear()
+            ax.plot(bestFits)
+            ax.set_title(f"Best Model: {bestFits[-1]:.2f} at Generation {(i+1)}")
+            ax.set_xlabel("Generations")
+            ax.set_ylabel("Fitness")
+            clear_output(wait=True) 
+            display(fig)    
+            #plt.show()        
+            plt.close(fig)
+            ckTime=time.perf_counter()        
 
         #paretoModels=paretoTournament(models)
         paretoModels=selectModels(models,elitismRate/100*popSize if elitismRate/100*popSize<len(models) else len(models))

{stackgp-0.0.5 → stackgp-0.0.7}/StackGP.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: StackGP
-Version: 0.0.5
+Version: 0.0.7
 Summary: A stack-based genetic programming system
 Author-email: Nathan Haut <hautnath@msu.edu>
 Project-URL: Homepage, https://github.com/hoolagans/StackGP

{stackgp-0.0.5 → stackgp-0.0.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "StackGP"
-version = "0.0.5"
+version = "0.0.7"
 authors = [
   { name="Nathan Haut", email="hautnath@msu.edu" },
 ]