package fehu

  1. Overview
  2. Docs
package fehu
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module Fehu_visualizeSource

Visualization sinks, encoders, and rollout helpers.

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.

Sourceval 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.

Sourceval 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.