mirror of
https://github.com/Azure/MachineLearningNotebooks.git
synced 2025-12-19 17:17:04 -05:00
18 lines
450 B
Python
18 lines
450 B
Python
'''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"])
|