-
Notifications
You must be signed in to change notification settings - Fork 1
/
dataset.pde
50 lines (38 loc) · 1.26 KB
/
dataset.pde
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
class Dataset {
// Every image consists of 784 (28*28) pixels
static final int DOODLE_PIXELS = 784;
Data[] trainingData;
Data[] testingData;
int sizeTrainingData;
int sizeTestingData;
int label;
Dataset(int sizeTrainingData, int sizeTestingData, byte[] data, int label) {
trainingData = new Data[sizeTrainingData];
testingData = new Data[sizeTestingData];
this.sizeTrainingData = sizeTrainingData;
this.sizeTestingData = sizeTestingData;
this.label = label;
prepareData(data);
}
Dataset(int sizeTrainingData, int sizeTestingData) {
trainingData = new Data[sizeTrainingData];
testingData = new Data[sizeTestingData];
this.sizeTrainingData = sizeTrainingData;
this.sizeTestingData = sizeTestingData;
}
void prepareData(byte[] data) {
int totalDataSize = sizeTrainingData + sizeTestingData;
for (int i = 0; i < totalDataSize; i++) {
int offset = i * DOODLE_PIXELS;
byte[] element = new byte[DOODLE_PIXELS];
for (int j = 0; j < DOODLE_PIXELS; j++) {
element[j] = data[j + offset];
}
if (i < sizeTrainingData) {
trainingData[i] = new Data(element, label);
} else {
testingData[i - sizeTrainingData] = new Data(element, label);
}
}
}
}