SHOGUN
3.2.1
Main Page
Related Pages
Modules
Classes
Files
File List
File Members
All
Classes
Namespaces
Files
Functions
Variables
Typedefs
Enumerations
Enumerator
Friends
Macros
Groups
Pages
src
shogun
clustering
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
14
#include <stdio.h>
15
#include <
shogun/lib/common.h
>
16
#include <
shogun/io/SGIO.h
>
17
#include <
shogun/distance/Distance.h
>
18
#include <
shogun/machine/DistanceMachine.h
>
19
20
namespace
shogun
21
{
22
class
CKMeansMiniBatchImpl
23
{
24
public
:
33
static
void
minibatch_KMeans
(int32_t k,
CDistance
*
distance
, int32_t batch_size, int32_t minib_iter,
SGMatrix<float64_t>
mus);
34
35
private
:
36
/* choose b integers between 0 and num-1
37
*
38
*/
39
static
SGVector<int32_t>
mbchoose_rand(int32_t b, int32_t num);
40
};
41
}
42
#endif
SHOGUN
Machine Learning Toolbox - Documentation