From c9f8ab18e4b73fb641df76e4134802eee713d735 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Wed, 14 Jan 2015 19:53:18 +0100
Subject: [PATCH] Added the PyBobLearnMiscKMeansTrainerObject
---
bob/learn/misc/main.cpp | 4 +++-
bob/learn/misc/main.h | 11 +++++++++++
2 files changed, 14 insertions(+), 1 deletion(-)
diff --git a/bob/learn/misc/main.cpp b/bob/learn/misc/main.cpp
index 8b683b3..fb8b2a5 100644
--- a/bob/learn/misc/main.cpp
+++ b/bob/learn/misc/main.cpp
@@ -43,7 +43,9 @@ static PyObject* create_module (void) {
if (!init_BobLearnMiscGaussian(module)) return 0;
if (!init_BobLearnMiscGMMStats(module)) return 0;
if (!init_BobLearnMiscGMMMachine(module)) return 0;
- if (!init_BobLearnMiscKMeansMachine(module)) return 0;
+ if (!init_BobLearnMiscKMeansMachine(module)) return 0;
+ if (!init_BobLearnMiscKMeansTrainer(module)) return 0;
+
static void* PyBobLearnMisc_API[PyBobLearnMisc_API_pointers];
diff --git a/bob/learn/misc/main.h b/bob/learn/misc/main.h
index 2f727f3..08d3919 100644
--- a/bob/learn/misc/main.h
+++ b/bob/learn/misc/main.h
@@ -105,5 +105,16 @@ bool init_BobLearnMiscKMeansMachine(PyObject* module);
int PyBobLearnMiscKMeansMachine_Check(PyObject* o);
+// KMeansTrainer
+typedef struct {
+ PyObject_HEAD
+ boost::shared_ptr<bob::learn::misc::KMeansTrainer> cxx;
+} PyBobLearnMiscKMeansTrainerObject;
+
+extern PyTypeObject PyBobLearnMiscKMeansTrainer_Type;
+bool init_BobLearnMiscKMeansTrainer(PyObject* module);
+int PyBobLearnMiscKMeansTrainer_Check(PyObject* o);
+
+
#endif // BOB_LEARN_EM_MAIN_H
--
GitLab