fraud-training.clj

(defn train
  "Trains network for :epoch-count number of epochs"
  []
  (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)))