本文整理汇总了Python中worker.Worker._policy_net_predict方法的典型用法代码示例。如果您正苦于以下问题：Python Worker._policy_net_predict方法的具体用法？Python Worker._policy_net_predict怎么用？Python Worker._policy_net_predict使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类worker.Worker的用法示例。
在下文中一共展示了Worker._policy_net_predict方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: testPolicyNetPredict
# 需要导入模块: from worker import Worker [as 别名]
# 或者: from worker.Worker import _policy_net_predict [as 别名]
def testPolicyNetPredict(self):
w = Worker(
name="test",
env=make_env(),
policy_net=self.global_policy_net,
value_net=self.global_value_net,
global_counter=self.global_counter,
discount_factor=self.discount_factor)
with self.test_session() as sess:
sess.run(tf.initialize_all_variables())
state = self.sp.process(self.env.reset())
processed_state = atari_helpers.atari_make_initial_state(state)
action_values = w._policy_net_predict(processed_state, sess)
self.assertEqual(action_values.shape, (4,))