rowan.a.tl/propeller
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #30 from dafg05/custom-report

Browse Source 
Add support for a custom-report function
This commit is contained in:
Lee Spector 2021-08-01 21:04:03 -04:00 committed by GitHub
commit 8a12ede981
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/propeller/gp.cljc
View File

@ -31,7 +31,7 @@
(defn gp
"Main GP loop."
[{:keys [population-size max-generations error-function instructions
max-initial-plushy-size]
max-initial-plushy-size custom-report]
:as argmap}]
;;
(prn {:starting-args (update (update argmap :error-function str) :instructions str)})
@ -49,7 +49,9 @@
(partial error-function argmap (:training-data argmap))
population))
best-individual (first evaluated-pop)]
(report evaluated-pop generation argmap)
(if (custom-report argmap)
((custom-report argmap) evaluated-pop generation argmap)
(report evaluated-pop generation argmap))
(cond
;; Success on training cases is verified on testing cases
(zero? (:total-error best-individual))