File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -554,13 +554,13 @@ def test_sample_by_noncrisp_observation_probabilities_mapping(self):
554
554
hmm = HiddenMarkovModel (tmat , obs )
555
555
# dtraj halfway-split between states 0 and 1
556
556
dtrajs = np .repeat ([0 , 1 ], 10 )
557
- n_samples = 300000
557
+ n_samples = 500000
558
558
samples = hmm .sample_by_observation_probabilities (dtrajs , n_samples )
559
559
# test that both hidden states map to correct distributions
560
560
probs_hidden1 = np .histogram (dtrajs [samples [0 ][:, 1 ]], bins = 2 )[0 ] / n_samples
561
561
probs_hidden2 = np .histogram (dtrajs [samples [1 ][:, 1 ]], bins = 2 )[0 ] / n_samples
562
- assert_array_almost_equal (probs_hidden1 , [.9 , .1 ], decimal = 3 )
563
- assert_array_almost_equal (probs_hidden2 , [.4 , .6 ], decimal = 3 )
562
+ assert_array_almost_equal (probs_hidden1 , [.9 , .1 ], decimal = 2 )
563
+ assert_array_almost_equal (probs_hidden2 , [.4 , .6 ], decimal = 2 )
564
564
565
565
def test_simulate_HMSM (self ):
566
566
hmsm = self .hmm_lag10_largest
You can’t perform that action at this time.
0 commit comments