def _summarize1(pred, tgt, loss, mask, suffix):
alf.summary.scalar(
"explained_variance" + suffix,
tensor_utils.explained_variance(pred, tgt, mask))
safe_mean_hist_summary('predict' + suffix, pred, mask)
safe_mean_hist_summary('target' + suffix, tgt, mask)
safe_mean_summary("loss" + suffix, loss, mask)
def _summary():
with self.name_scope:
tf.summary.scalar("values", tf.reduce_mean(info.critic))
tf.summary.scalar("returns", tf.reduce_mean(info.returns))
safe_mean_hist_summary("td_error", info.returns - info.critic)
tf.summary.scalar(
"explained_variance_of_return_by_value",
common.explained_variance(info.critic, info.returns))
def _summarize(v, r, td, suffix):
alf.summary.scalar(
"explained_variance_of_return_by_value" + suffix,
tensor_utils.explained_variance(v, r, mask))
safe_mean_hist_summary('values' + suffix, v, mask)
safe_mean_hist_summary('returns' + suffix, r, mask)
safe_mean_hist_summary("td_error" + suffix, td, mask)