From 5c43dbbc232badb7e9990cd4cc161a6a5c5f57a8 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Tue, 31 May 2016 18:40:28 +0200
Subject: [PATCH] Fixed issue #18. Corrected the prints for the KMeans machine
---
bob/learn/em/train.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/bob/learn/em/train.py b/bob/learn/em/train.py
index e5c2f07..81a41cb 100644
--- a/bob/learn/em/train.py
+++ b/bob/learn/em/train.py
@@ -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)
--
GitLab