mo8it/ReCo.jl
1
0
Fork 0
mirror of https://gitlab.rlp.net/mobitar/ReCo.jl.git synced 2024-09-19 19:01:17 +00:00
Code Issues Releases Activity

export NearestNeighbourEnv

Browse source
This commit is contained in:
Mo8it 2022-01-31 01:51:23 +01:00
parent 9e3b71197c
commit e8a5d4f684
2 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/RL/RL.jl
View file

@ -1,6 +1,6 @@
module RL
export run_rl, LocalCOMWithAdditionalShapeRewardEnv, OriginEnv
export run_rl, LocalCOMWithAdditionalShapeRewardEnv, OriginEnv, NearestNeighbourEnv
using Base: OneTo
@ -29,8 +29,8 @@ include("Reward.jl")
function gen_agent(
n_states::Int64, n_actions::Int64, ϵ_stable::Float64, reward_discount::Float64
)
# TODO: Optimize warmup and decay
warmup_steps = 500_000
# TODO: Optimize warming up and decay
warmup_steps = 400_000
decay_steps = 5_000_000
policy = QBasedPolicy(;
@ -188,5 +188,6 @@ end
include("Envs/LocalCOMWithAdditionalShapeRewardEnv.jl")
include("Envs/OriginEnv.jl")
include("Envs/NearestNeighbourEnv.jl")
end # module

8
src/ReCo.jl
View file

@ -1,7 +1,13 @@
module ReCo
export init_sim,
run_sim, run_rl, animate, plot_snapshot, LocalCOMWithAdditionalShapeRewardEnv, OriginEnv
run_sim,
run_rl,
animate,
plot_snapshot,
LocalCOMWithAdditionalShapeRewardEnv,
OriginEnv,
NearestNeighbourEnv
using StaticArrays: SVector
using JLD2: JLD2