एजेंट TensorFlow में सुदृढीकरण सीखने के लिए एक पुस्तकालय है।

import tensorflow as tf
from tf_agents.networks import q_network
from tf_agents.agents.dqn import dqn_agent

q_net = q_network.QNetwork(
  train_env.observation_spec(),
  train_env.action_spec(),
  fc_layer_params=(100,))

agent = dqn_agent.DqnAgent(
  train_env.time_step_spec(),
  train_env.action_spec(),
  q_network=q_net,
  optimizer=optimizer,
  td_errors_loss_fn=common.element_wise_squared_loss,
  train_step_counter=tf.Variable(0))

agent.initialize()
1नोटबुक में चलाएं4
TF-Agents अच्छी तरह से परीक्षण किए गए मॉड्यूलर घटकों को प्रदान करके नए RL एल्गोरिदम को डिजाइन करना, कार्यान्वित करना और परीक्षण करना आसान बनाता है जिसे संशोधित और विस्तारित किया जा सकता है। यह अच्छे परीक्षण एकीकरण और बेंचमार्किंग के साथ तेज़ कोड पुनरावृत्ति को सक्षम बनाता है।