SHOGUN  3.2.1
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
KMeansMiniBatchImpl.h
Go to the documentation of this file.
1 /*
2  * This program is free software; you can redistribute it and/or modify
3  * it under the terms of the GNU General Public License as published by
4  * the Free Software Foundation; either version 3 of the License, or
5  * (at your option) any later version.
6  *
7  * Written (W) 2014 Parijat Mazumdar
8  */
9 
10 #ifndef _MBKMEANS_H__
11 #define _MBKMEANS_H__
12 
13 #include <shogun/lib/config.h>
14 
15 #include <stdio.h>
16 #include <shogun/lib/common.h>
17 #include <shogun/io/SGIO.h>
20 
21 namespace shogun
22 {
24 {
25  public:
34  static void minibatch_KMeans(int32_t k, CDistance* distance, int32_t batch_size, int32_t minib_iter, SGMatrix<float64_t> mus);
35 
36  private:
37  /* choose b integers between 0 and num-1
38  *
39  */
40  static SGVector<int32_t> mbchoose_rand(int32_t b, int32_t num);
41 };
42 }
43 #endif

SHOGUN Machine Learning Toolbox - Documentation