;; ardegazu-train-kit — the learning half of an ardegazu.ro bot.
;;
;; The package root: one var per public symbol, re-exported by the :index
;; module as dist/index.js (the per-subpath modules re-export the same vars
;; from their own namespaces).
(ns ardegazu.train.index
  (:require [ardegazu.train.model :as model]
            [ardegazu.train.rev :as rev]
            [ardegazu.train.forward :as forward]
            [ardegazu.train.episodes :as episodes]
            [ardegazu.train.checkpoint :as checkpoint]
            [ardegazu.train.tabular :as tabular]
            [ardegazu.train.dqn :as dqn]
            [ardegazu.train.tf :as tf]
            [ardegazu.train.gate :as gate]
            [ardegazu.train.ingest :as ingest]))

(def is-mlp-model model/is-mlp-model)
(def is-tabular-model model/is-tabular-model)
(def is-model model/is-model)
(def pick-action model/pick-action)

(def model-rev rev/model-rev)
(def model-version rev/model-version)

(def mlp-forward forward/mlp-forward)
(def mlp-init forward/mlp-init)

(def EpisodeWriter episodes/EpisodeWriter)
(def read-new-transitions episodes/read-new-transitions)

(def checkpoint-path checkpoint/checkpoint-path)
(def candidate-path checkpoint/candidate-path)
(def load-model checkpoint/load-model)
(def save-model checkpoint/save-model)
(def CheckpointStore checkpoint/CheckpointStore)

(def ensure-row tabular/ensure-row)
(def q-update tabular/q-update)

(def net-qs dqn/net-qs)
(def dqn-targets dqn/dqn-targets)
(def cap-replay dqn/cap-replay)

(def load-tf tf/load-tf)
(def net-to-tf tf/net-to-tf)
(def tf-to-net tf/tf-to-net)
(def fit-net tf/fit-net)

(def resolve-model gate/resolve-model)
(def gate-promotion gate/gate-promotion)

(def expand-dirs ingest/expand-dirs)
(def ingest-state-path ingest/ingest-state-path)
(def load-ingest-state ingest/load-ingest-state)
(def save-ingest-state ingest/save-ingest-state)