forked from jacquelineCelia/hdphmm
-
Notifications
You must be signed in to change notification settings - Fork 0
/
randomizer.cc
69 lines (61 loc) · 1.85 KB
/
randomizer.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
/* -*- C++ -*-
*
* Copyright (c) 2014
* Spoken Language Systems Group
* MIT Computer Science and Artificial Intelligence Laboratory
* Massachusetts Institute of Technology
*
* All Rights Reserved
./randomizer.cc
* FILE: cluster.cc *
* *
* *
* Chia-ying (Jackie) Lee <chiaying@csail.mit.edu> *
* Feb 2014 *
*********************************************************************/
#include "randomizer.h"
using namespace std;
Randomizer::Randomizer() {
_uniformer = new Uniformer(3000000);
}
void Randomizer::LoadData(vector<Datum*>& data) {
_data = data;
_index = _data.size();
}
void Randomizer::Shuffle() {
// shuffle _data.begin() and _data.begin() + _index
for (int i = 0; i < _index; ++i) {
int j = _uniformer -> GetDiscreteSample() % _index;
Datum* a = _data[i];
_data[i] = _data[j];
_data[j] = a;
}
}
vector<Datum*> Randomizer::GetBatch(int batch_size) {
batch_size = batch_size > (int) _data.size() ? _data.size() : batch_size;
if (_index + batch_size - 1 >= (int) _data.size()) {
// Randomize();
_index = 0;
}
vector<Datum*> batch(_data.begin() + _index, _data.begin() + _index + batch_size);
_index += batch_size;
return batch;
}
void Randomizer::Randomize() {
vector<Datum*> copy_data;
// Shuffle();
int ptr = 0;
if (_index < (int) _data.size()) {
copy_data.assign(_data.begin() + _index, _data.end());
ptr = copy_data.size();
}
copy_data.resize(_data.size());
copy(_data.begin(), _data.begin() + _index, copy_data.begin() + ptr);
_data = copy_data;
_index = 0;
}
Randomizer::~Randomizer() {
if (_uniformer != NULL) {
delete _uniformer;
}
}