mo8it/ReCo.jl
1
0
Fork 0
mirror of https://gitlab.rlp.net/mobitar/ReCo.jl.git synced 2025-09-02 09:02:35 +00:00
Code Issues Releases Activity

Added reward shaping plot

Browse source
This commit is contained in:
Mo8it 2022-01-30 04:38:57 +01:00
parent eb29ca3d10
commit fd83c13333
5 changed files with 35 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
analysis/mean_squared_displacement.jl
View file

@ -133,7 +133,7 @@ function plot_mean_sq_displacement_with_expectation(
fig = gen_figure()
ax = Axis(fig[1, 1]; xlabel=L"t", ylabel=L"\mathbf{MSD}", xscale=log10, yscale=log10)
ax = Axis(fig[1, 1]; xlabel=L"t", ylabel=L"\mathbf{MSD}(t)", xscale=log10, yscale=log10)
t_linrange = LinRange(ts[1], ts[end], 1000)

2
analysis/radial_distribution_function/radial_distribution_function.jl
View file

@ -164,7 +164,7 @@ function plot_radial_distributions(;
xticks=0:(2 * particle_radius):floor(Int64, max_lower_radius),
yticks=0:ceil(Int64, max_g),
xlabel=L"r / d",
ylabel=L"g",
ylabel=L"g(r)",
limits=(min_lower_radius - 0.03, max_lower_radius + 0.03, min_g, max_g * 1.03),
)

2
analysis/reward_discount_analysis.jl
View file

@ -14,7 +14,7 @@ function run_reward_discount_analysis()
γ = γs[γ_ind]
env_helper = ReCo.run_rl(;
EnvType=ReCo.OriginEnv,
n_episodes=3,
n_episodes=400,
episode_duration=15.0,
n_particles=150,
update_actions_at=0.08,