12 lines
525 B
Python
12 lines
525 B
Python
from recsim.agents import full_slate_q_agent
|
|
|
|
def create_agent_helper(agent=full_slate_q_agent.FullSlateQAgent, **kwargs):
|
|
def create_agent(sess, environment, eval_mode, summary_writer=None):
|
|
print(f"using {agent.__name__}")
|
|
kwargs['observation_space'] = environment.observation_space
|
|
kwargs['action_space'] = environment.action_space
|
|
kwargs['summary_writer'] = summary_writer
|
|
kwargs['eval_mode'] = eval_mode
|
|
|
|
return agent(sess, **kwargs)
|
|
return create_agent |