def test_hmm_weights():
simple_weights = viterbi.get_HMM_weights('tests/hmm_simple.dat')
assert_almost_equals(1.0, np.exp(simple_weights[('D', START_TAG, TRANS)]), places=3)
assert_almost_equals(1.0, np.exp(simple_weights[('N', 'D', TRANS)]), places=3)
assert_almost_equals(1.0, np.exp(simple_weights[('V', 'N', TRANS)]), places=3)
assert_almost_equals(1.0, np.exp(simple_weights[('JJ', 'V', TRANS)]), places=3)
assert_almost_equals(1.0, np.exp(simple_weights[(END_TAG, 'JJ', TRANS)]), places=3)
def setup_module ():
global defined_weights
global alltags
global hmm_weights
start_tag = constants.START_TAG
trans = constants.TRANS
end_tag = constants.END_TAG
emit = constants.EMIT
hmm_weights = viterbi.get_HMM_weights(TRAIN_FILE)
defined_weights = {('N','they',emit):-1,('N','can',emit):-3,('N','fish',emit):-3,\
('V','they',emit):-10,('V','can',emit):-2,('V','fish',emit):-3,\
('N','N',trans):-5,('V','N',trans):-2,(end_tag,'N',trans):-3,\
('N','V',trans):-1,('V','V',trans):-4,(end_tag,'V',trans):-3,\
('N',start_tag,trans):-1,('V',start_tag,trans):-1}
alltags = preproc.getAllTags(TRAIN_FILE)
def setup_module():
global defined_weights
global alltags
global hmm_weights
start_tag = constants.START_TAG
trans = constants.TRANS
end_tag = constants.END_TAG
emit = constants.EMIT
hmm_weights = viterbi.get_HMM_weights(TRAIN_FILE)
defined_weights = {('N','they',emit):-1,('N','can',emit):-3,('N','fish',emit):-3,\
('V','they',emit):-10,('V','can',emit):-2,('V','fish',emit):-3,\
('N','N',trans):-5,('V','N',trans):-2,(end_tag,'N',trans):-3,\
('N','V',trans):-1,('V','V',trans):-4,(end_tag,'V',trans):-3,\
('N',start_tag,trans):-1,('V',start_tag,trans):-1}
alltags = preproc.getAllTags(TRAIN_FILE)
def test_hmm_weights():
simple_weights = viterbi.get_HMM_weights('tests/hmm_simple.dat')
assert_almost_equals(1.0,
np.exp(simple_weights[('D', START_TAG, TRANS)]),
places=3)
assert_almost_equals(1.0,
np.exp(simple_weights[('N', 'D', TRANS)]),
places=3)
assert_almost_equals(1.0,
np.exp(simple_weights[('V', 'N', TRANS)]),
places=3)
assert_almost_equals(1.0,
np.exp(simple_weights[('JJ', 'V', TRANS)]),
places=3)
assert_almost_equals(1.0,
np.exp(simple_weights[(END_TAG, 'JJ', TRANS)]),
places=3)
