Added a plot function for the performance tracker

ah-box · Oct 23, 2016 · f7c7eec · f7c7eec
1 parent 5b4e4cd
commit f7c7eec
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/NAMESPACE b/NAMESPACE
@@ -1,6 +1,7 @@
 # Generated by roxygen2: do not edit by hand
 
 S3method(plot,NEATFormulaSimulation)
+S3method(plot,NEATSimulation)
 S3method(plot,rneatneuralnet)
 export(NEATSimulation.GetStateHistoryForGenomeAndSpecies)
 export(NEATSimulation.RunSingleGeneration)

diff --git a/R/neat.R b/R/neat.R
@@ -1037,6 +1037,10 @@ NEATSimulation.GetStateHistoryForGenomeAndSpecies <- function(simulation, genome
 
 }
 
+#' @export
+plot.NEATSimulation <- function(simulation){
+  plotPerformanceTracker(simulation$PerformanceTracker)
+}
 
 plotPerformanceTracker <- function(data){
   plot(x=data[,"generation"],y=data[,"maxFitness"],col="blue",main="Fitness",xlab="Generation",ylab="Fitness",type="o",ylim=c(0,max(data[,"maxFitness"])),lwd=2)