def task(self):
data_filename = "robot_no_momentum.data"
hmm, d = train_hmm_from_data(data_filename)
err_full = run_viterbi(hmm, d)
data_filename_m = "robot_with_momentum.data"
hmm_m, d_m = train_hmm_from_data(data_filename_m)
err_full_m = run_viterbi(hmm_m, d_m)
listNames = ["Without momentum", "With momentum"]
listData = [1 - err_full, 1 - err_full_m]
chart = {"chart": {"defaultSeriesType": "column"},
"xAxis": {"categories": listNames},
"yAxis": {"title": {"text": "Fraction Correct"}},
"title": {"text": "HMM performance on"
" inferring robot location."},
"series": [{"name": "Test set performance",
"data": listData}]}
return chart
def test_small_robot_dataset(self):
data_filename = "robot_small.data"
data_filename = normalize_filename(data_filename)
hmm, d = train_hmm_from_data(data_filename)
err_full = run_viterbi(hmm, d, True)
self.assertAlmostEqual(err_full, 2.0 / 9)
