Carlos-Francisco Méndez-Cruz

Conditional Random Fields

......@@ -212,10 +212,11 @@ if __name__ == "__main__":
# Fixed parameters
crf = sklearn_crfsuite.CRF(
algorithm='lbfgs',
c1=0.1,
c2=0.1,
max_iterations=100,
all_possible_transitions=True
#c1=0.1,
#c2=0.1,
max_iterations=100
#,
#all_possible_transitions=True
)
# Original: labels = list(crf.classes_)
......@@ -237,9 +238,10 @@ if __name__ == "__main__":
oFile.write("********** TRAINING AND TESTING REPORT **********\n")
oFile.write("Training file: " + options.trainingFile + '\n')
oFile.write('\n')
oFile.write("Model: {}".format(crf))
#oFile.write('best params:' + str(crf.best_params_) + '\n')
oFile.write('best CV score:' + str(crf.best_score_) + '\n')
oFile.write('model size: {:0.2f}M\n'.format(crf.best_estimator_.size_ / 1000000))
#oFile.write('best CV score:' + str(crf.best_score_) + '\n')
#oFile.write('model size: {:0.2f}M\n'.format(crf.best_estimator_.size_ / 1000000))
print("Training done in: %fs" % (time() - t0))
t0 = time()
......