mo8it/ReCo.jl
1
0
Fork 0
mirror of https://gitlab.rlp.net/mobitar/ReCo.jl.git synced 2024-12-21 00:51:21 +00:00
Code Issues Releases Activity

Specify trajectory types

Browse source
This commit is contained in:
Mo8it 2022-01-11 18:43:43 +01:00
parent 0cf59e04a5
commit eba1e53cd2
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/RL/RL.jl
View file

@ -298,7 +298,11 @@ function gen_agent(n_states::Int64, n_actions::Int64, ϵ_stable::Float64)
),
)
return Agent(; policy=policy, trajectory=VectorSARTTrajectory())
trajectory = VectorSARTTrajectory(;
state=Int64, action=Int64, reward=Float64, terminal=Bool
)
return Agent(; policy=policy, trajectory=trajectory)
end
function run_rl(;