Skip to content
Snippets Groups Projects
Commit 4563d8c8 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Merge pull request #20 from tiagofrepereira2012/master

Fixed issue #18. Corrected the prints for the KMeans machine
parents 8dfe578d 5c43dbbc
No related branches found
No related tags found
No related merge requests found
......@@ -52,7 +52,12 @@ def train(trainer, machine, data, max_iterations = 50, convergence_threshold=Non
if hasattr(trainer,"compute_likelihood"):
average_output = trainer.compute_likelihood(machine)
logger.info("log likelihood = %f", average_output)
if type(machine) is bob.learn.em.KMeansMachine:
logger.info("average euclidean distance = %f", average_output)
else:
logger.info("log likelihood = %f", average_output)
convergence_value = abs((average_output_previous - average_output)/average_output_previous)
logger.info("convergence value = %f",convergence_value)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment