clojure-conj-3.clj

(def params
  {:optimizer         (adam/adam)
   :batch-size        100
   :epoch-count       50
   :epoch-size        200000})

(defn train
  []
  (let [network (network/linear-network network-description)
        [train-orig test-ds] (get-train-test-dataset)
        train-ds (experiment-util/infinite-class-balanced-dataset train-orig
                                            :class-key :label
                                            :epoch-size (:epoch-size params))]
    (experiment-train/train-n network train-ds test-ds
                                      :batch-size (:batch-size params)
                                      :epoch-count (:epoch-count params)
                                      :optimizer (:optimizer params)
                                      :test-fn f1-test-fn)))