SHOGUN
3.2.1
首页
相关页面
模块
类
文件
文件列表
文件成员
全部
类
命名空间
文件
函数
变量
类型定义
枚举
枚举值
友元
宏定义
组
页
src
shogun
clustering
KMeansMiniBatchImpl.h
浏览该文件的文档.
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
机器学习工具包 - 项目文档