Skip to content
/ rebar Public

Tools for prototyping in deep reinforcement learning

Notifications You must be signed in to change notification settings

csreid/rebar

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Rebar

Tools for experiments with reinforcement learning with Torch and Python, so that they can look like this:

env = gym.make('CartPole-v1')

Q_network = Sequential(
	Linear(4, 32),
	ReLU()
	Linear(32, 2)
)

agt_cfg = {
	"action_space": env.action_space,
	"observation_space": env.observation_space,
	"Q": Q
}

exp = Experiment(
	env,
	QLearner,
	agt_cfg,
	eval_freq=100,
	steps_per_trial=5000,
	n_trials=10
)

results = exp.run()

About

Tools for prototyping in deep reinforcement learning

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages