def _ext_identity(samples):
"""Extracts the values of all latent variables."""
res = collections.OrderedDict()
res['county_effect_mean'] = util.get_columns(
samples, r'^county_effect_mean$')[:, 0]
res['county_effect_scale'] = util.get_columns(
samples, r'^county_effect_scale$')[:, 0]
res['county_effect'] = util.get_columns(samples,
r'^county_effect\.\d+$')
res['weight'] = util.get_columns(samples, r'^weight\.\d+$')
res['log_radon_scale'] = util.get_columns(samples,
r'^log_radon_scale$')[:, 0]
return res
def _ext_identity(samples):
"""Extracts the values of all latent variables."""
res = collections.OrderedDict()
res['mean_log_volatility'] = util.get_columns(
samples, r'^mean_log_volatility$')
res['white_noise_shock_scale'] = util.get_columns(
samples, r'^white_noise_shock_scale$')
res['persistence_of_volatility'] = util.get_columns(
samples, r'^persistence$')
res['log_volatility'] = util.get_columns(
samples,
r'^log_volatilities\.\d+$',
)
return res
def _ext_identity(samples):
"""Extract all the parameters."""
res = collections.OrderedDict()
res['amplitude'] = util.get_columns(
samples,
r'^amplitude$',
)[:, 0]
res['length_scale'] = util.get_columns(
samples,
r'^length_scale$',
)[:, 0]
res['log_intensity'] = util.get_columns(
samples,
r'^log_intensity\.\d+$',
)
return res
def _ext_identity(samples):
"""Extract all the parameters."""
res = collections.OrderedDict()
res['unscaled_weights'] = util.get_columns(
samples,
r'^unscaled_weights\.\d+$',
)
res['local_scales'] = util.get_columns(
samples,
r'^local_scales\.\d+$',
)
res['global_scale'] = util.get_columns(
samples,
r'^global_scale$',
)[:, 0]
return res
def _ext_identity(samples):
"""Extracts all the parameters."""
res = collections.OrderedDict()
res['mean_student_ability'] = util.get_columns(
samples,
r'^mean_student_ability$',
)[:, 0]
res['student_ability'] = util.get_columns(
samples,
r'^student_ability\.\d+$',
)
res['question_difficulty'] = util.get_columns(
samples,
r'^question_difficulty\.\d+$',
)
return res
def _ext_per_example_test_nll(samples):
return util.get_columns(samples, r'^per_example_test_nll\.\d+$')
def _ext_test_nll(samples):
return util.get_columns(samples, r'^test_nll$')[:, 0]
def _ext_identity(samples):
return util.get_columns(samples, r'^weights\.\d+$')
def _ext_identity(samples):
"""Extracts the values of all latent variables."""
locs = util.get_columns(samples, r'^loc\.\d+$')
return locs
