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

Add support for a custom-report function

Browse Source
This commit is contained in:
Daniel Flores García 2021-07-31 14:01:25 -04:00
parent 4058b8e7b0
commit f25cd75f2a
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))