Fehu_visualize (p.fehu.1.0.0~alpha2.doc.fehu.visualize.Fehu

Sourcemodule Overlay : sig ... end

Overlays for augmenting rendered frames with diagnostic annotations.

Sourcemodule Video : sig ... end

Video recording wrappers for environments and vectorized environments.

Sourcemodule Sink : sig ... end

Sourceval push : Sink.t -> Fehu.Render.t -> unit

Push a single frame to the sink.

Sourceval push_many : Sink.t -> Fehu.Render.t array -> unit

Push multiple frames sequentially.

Source

val record_rollout : 
  env:('obs, 'act, Fehu.Render.t) Fehu.Env.t ->
  policy:('obs -> 'act * float option * float option) ->
  steps:int ->
  ?overlay:Overlay.t ->
  sink:Sink.t ->
  unit ->
  unit

Record a rollout from a single environment to the sink.

Source

val record_evaluation : 
  vec_env:('obs, 'act, Fehu.Render.t) Fehu.Vector_env.t ->
  policy:('obs array -> 'act array * float array option * float array option) ->
  n_episodes:int ->
  ?max_steps:int ->
  layout:[ `Single_each | `NxM_grid of int * int ] ->
  ?overlay:Overlay.t ->
  sink:Sink.t ->
  unit ->
  Fehu.Training.stats

Evaluate a vectorized policy while recording frames.