Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
39 lines (27 sloc) 763 Bytes
open Core
type t
val create
: ?epsilon_init:float
-> hidden_layer_sizes:int list
-> activation:[ `Sigmoid | `Relu ]
-> representation:[ `Original | `Modified | `Expanded ]
-> unit
-> t
val representation : t -> [ `Original | `Modified | `Expanded ]
val eval : t -> Equity.Setup.t array -> float array
module Setup : sig
type t [@@deriving sexp]
val create : Equity.Setup.t -> [ `Original | `Modified | `Expanded ] -> t
module And_valuation : sig
type nonrec t = t * float [@@deriving sexp]
end
end
val train
: t
-> (Setup.t * float) Replay_memory.t
-> minibatch_size:int
-> minibatches_number:int
-> unit
val save : t -> filename:string -> unit
val load : t -> filename:string -> unit
val sexp_of_vars : t -> Sexp.t
You can’t perform that action at this time.