def hmm_train(results):
train_file = results.train
freq_file = results.freq
logger.debug( 'Started training HMM with options:' + "\n" +
'training file: ' + str(train_file) + "\n" +
'frequency file:' + str(freq_file) + "\n")
if not os.path.exists('model/hmm-model'):
classifier = HMM()
classifier.train(train_file,freq_file)
logger.info("Done Training, model is written in model file")
model = classifier.get_theta()
write_obj(model, 'hmm-model')
else:
logger.info('model already exists, nothing to do!')
