jprdonnelly/MachineLearningNotebooks
1
0
Fork 0
mirror of https://github.com/Azure/MachineLearningNotebooks.git synced 2025-12-19 17:17:04 -05:00
Code Issues Projects Releases Wiki Activity

update samples from Release-11 as a part of 1.5.0 SDK stable release

Browse Source
This commit is contained in:
vizhur
2020-05-11 18:59:22 +00:00
parent e98131729e
commit 553fa43e17
76 changed files with 3818 additions and 622 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
how-to-use-azureml/reinforcement-learning/cartpole-on-single-compute/files/utils/callbacks.py Normal file
View File

@@ -0,0 +1,17 @@
'''RLlib callbacks module:
Common callback methods to be passed to RLlib trainer.
'''
from azureml.core import Run
def on_train_result(info):
'''Callback on train result to record metrics returned by trainer.
'''
run = Run.get_context()
run.log(
name='episode_reward_mean',
value=info["result"]["episode_reward_mean"])
run.log(
name='episodes_total',
value=info["result"]["episodes_total"])