def __init__(self, use_markov_chain=False):
dataset = data.brownian_motion_missing_middle_observations()
super(BrownianMotionMissingMiddleObservations, self).__init__(
name='brownian_motion_missing_middle_observations',
pretty_name='Brownian Motion Missing Middle Observations',
use_markov_chain=use_markov_chain,
**dataset)
def brownian_motion_missing_middle_observations():
"""Brownian Motion with missing observations.
Returns:
target: StanModel.
"""
dataset = data.brownian_motion_missing_middle_observations()
return brownian_motion.brownian_motion(**dataset)
def brownian_motion_unknown_scales_missing_middle_observations():
"""Brownian Motion with missing observations and unknown scale parameters.
Returns:
target: StanModel.
"""
dataset = data.brownian_motion_missing_middle_observations()
return brownian_motion.brownian_motion_unknown_scales(locs=dataset['locs'])
def __init__(self, use_markov_chain=False):
dataset = data.brownian_motion_missing_middle_observations()
del dataset['innovation_noise_scale']
del dataset['observation_noise_scale']
super(BrownianMotionUnknownScalesMissingMiddleObservations, self).__init__(
name='brownian_motion_unknown_scales_missing_middle_observations',
pretty_name='Brownian Motion with Unknown Scales',
use_markov_chain=use_markov_chain,
**dataset)
def __init__(self):
dataset = data.brownian_motion_missing_middle_observations()
super(BrownianMotionMissingMiddleObservations, self).__init__(
name='brownian_motion_missing_middle_observations',
pretty_name='Brownian Motion Missing Middle Observations',
**dataset)
def testBrownianMotion(self): dataset = data.brownian_motion_missing_middle_observations() self.assertEqual((30,), dataset['observed_locs'].shape)
