rlberry.utils.check_fit_additive

rlberry.utils.check_fit_additive(agent, env='continuous_state', init_kwargs=None)

Check that fitting two times with 10 fit budget is the same as fitting one time with 20 fit budget.

Parameters:

agent: rlberry agent module

Agent class to test.

env: tuple (env_ctor, env_kwargs) or str in [“continuous_state”, “discrete_state”], default=”continuous_state”

if tuple, env is the constructor and keywords of the env on which to test. if str in [“continuous_state”, “discrete_state”], we use a default Benchmark environment.

init_kwargsdict

Arguments required by the agent’s constructor.