diff --git a/src/propeller/gp.cljc b/src/propeller/gp.cljc index 1ad3f71..e5dccf6 100755 --- a/src/propeller/gp.cljc +++ b/src/propeller/gp.cljc @@ -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))