diff --git a/src/RL/RL.jl b/src/RL/RL.jl index dc3e029..011b61b 100644 --- a/src/RL/RL.jl +++ b/src/RL/RL.jl @@ -19,6 +19,7 @@ using Random: Random using ProgressMeter: ProgressMeter using JLD2: JLD2 using LaTeXStrings: LaTeXStrings, @L_str +using UnicodePlots: UnicodePlots using ..ReCo: ReCo @@ -32,7 +33,7 @@ const DEFAULT_UPDATE_ACTIONS_AT = 0.1 const DEFAULT_N_PARTICLES = 100 const DEFAULT_SEED = 42 const DEFAULT_ϵ_STABLE = 0.00001 -const DEFAULT_PACKING_FRACTION = 0.01 +const DEFAULT_PACKING_FRACTION = 0.02 const DEFAULT_SHOW_PROGRESS = true const DEFAULT_SHOW_SIMULATION_PROGRESS = true const DEFAULT_REWARD_DISCOUNT = 0.1 @@ -202,7 +203,12 @@ function run_rl( JLD2.save_object(env_helper_path, env_helper) end - ProgressMeter.next!(progress; showvalues=[(:rewards, hook.rewards)]) + ProgressMeter.next!( + progress; + showvalues=[( + :R, repr(UnicodePlots.lineplot(1:episode, hook.rewards; border=:none)) + )], + ) end # Post experiment