summaryrefslogtreecommitdiff
path: root/chromium/media/learning/impl/fisher_iris_dataset.cc
blob: 00bac50342d036caf3571a5158416e9796ce9a22 (plain)
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "media/learning/impl/fisher_iris_dataset.h"

#include <vector>

namespace {
struct IrisExample : public media::learning::LabelledExample {
  IrisExample(float sepal_length,
              float sepal_width,
              float petal_length,
              float petal_width,
              const char* class_name) {
    target_value = media::learning::TargetValue(class_name);
    features.push_back(
        media::learning::FeatureValue(static_cast<int>(sepal_length * 10)));
    features.push_back(
        media::learning::FeatureValue(static_cast<int>(sepal_width * 10)));
    features.push_back(
        media::learning::FeatureValue(static_cast<int>(petal_length * 10)));
    features.push_back(
        media::learning::FeatureValue(static_cast<int>(petal_width * 10)));
  }
};

std::vector<IrisExample> iris_examples({
    {5.1, 3.5, 1.4, 0.2, "Iris-setosa"},
    {4.9, 3.0, 1.4, 0.2, "Iris-setosa"},
    {4.7, 3.2, 1.3, 0.2, "Iris-setosa"},
    {4.6, 3.1, 1.5, 0.2, "Iris-setosa"},
    {5.0, 3.6, 1.4, 0.2, "Iris-setosa"},
    {5.4, 3.9, 1.7, 0.4, "Iris-setosa"},
    {4.6, 3.4, 1.4, 0.3, "Iris-setosa"},
    {5.0, 3.4, 1.5, 0.2, "Iris-setosa"},
    {4.4, 2.9, 1.4, 0.2, "Iris-setosa"},
    {4.9, 3.1, 1.5, 0.1, "Iris-setosa"},
    {5.4, 3.7, 1.5, 0.2, "Iris-setosa"},
    {4.8, 3.4, 1.6, 0.2, "Iris-setosa"},
    {4.8, 3.0, 1.4, 0.1, "Iris-setosa"},
    {4.3, 3.0, 1.1, 0.1, "Iris-setosa"},
    {5.8, 4.0, 1.2, 0.2, "Iris-setosa"},
    {5.7, 4.4, 1.5, 0.4, "Iris-setosa"},
    {5.4, 3.9, 1.3, 0.4, "Iris-setosa"},
    {5.1, 3.5, 1.4, 0.3, "Iris-setosa"},
    {5.7, 3.8, 1.7, 0.3, "Iris-setosa"},
    {5.1, 3.8, 1.5, 0.3, "Iris-setosa"},
    {5.4, 3.4, 1.7, 0.2, "Iris-setosa"},
    {5.1, 3.7, 1.5, 0.4, "Iris-setosa"},
    {4.6, 3.6, 1.0, 0.2, "Iris-setosa"},
    {5.1, 3.3, 1.7, 0.5, "Iris-setosa"},
    {4.8, 3.4, 1.9, 0.2, "Iris-setosa"},
    {5.0, 3.0, 1.6, 0.2, "Iris-setosa"},
    {5.0, 3.4, 1.6, 0.4, "Iris-setosa"},
    {5.2, 3.5, 1.5, 0.2, "Iris-setosa"},
    {5.2, 3.4, 1.4, 0.2, "Iris-setosa"},
    {4.7, 3.2, 1.6, 0.2, "Iris-setosa"},
    {4.8, 3.1, 1.6, 0.2, "Iris-setosa"},
    {5.4, 3.4, 1.5, 0.4, "Iris-setosa"},
    {5.2, 4.1, 1.5, 0.1, "Iris-setosa"},
    {5.5, 4.2, 1.4, 0.2, "Iris-setosa"},
    {4.9, 3.1, 1.5, 0.1, "Iris-setosa"},
    {5.0, 3.2, 1.2, 0.2, "Iris-setosa"},
    {5.5, 3.5, 1.3, 0.2, "Iris-setosa"},
    {4.9, 3.1, 1.5, 0.1, "Iris-setosa"},
    {4.4, 3.0, 1.3, 0.2, "Iris-setosa"},
    {5.1, 3.4, 1.5, 0.2, "Iris-setosa"},
    {5.0, 3.5, 1.3, 0.3, "Iris-setosa"},
    {4.5, 2.3, 1.3, 0.3, "Iris-setosa"},
    {4.4, 3.2, 1.3, 0.2, "Iris-setosa"},
    {5.0, 3.5, 1.6, 0.6, "Iris-setosa"},
    {5.1, 3.8, 1.9, 0.4, "Iris-setosa"},
    {4.8, 3.0, 1.4, 0.3, "Iris-setosa"},
    {5.1, 3.8, 1.6, 0.2, "Iris-setosa"},
    {4.6, 3.2, 1.4, 0.2, "Iris-setosa"},
    {5.3, 3.7, 1.5, 0.2, "Iris-setosa"},
    {5.0, 3.3, 1.4, 0.2, "Iris-setosa"},
    {7.0, 3.2, 4.7, 1.4, "Iris-versicolor"},
    {6.4, 3.2, 4.5, 1.5, "Iris-versicolor"},
    {6.9, 3.1, 4.9, 1.5, "Iris-versicolor"},
    {5.5, 2.3, 4.0, 1.3, "Iris-versicolor"},
    {6.5, 2.8, 4.6, 1.5, "Iris-versicolor"},
    {5.7, 2.8, 4.5, 1.3, "Iris-versicolor"},
    {6.3, 3.3, 4.7, 1.6, "Iris-versicolor"},
    {4.9, 2.4, 3.3, 1.0, "Iris-versicolor"},
    {6.6, 2.9, 4.6, 1.3, "Iris-versicolor"},
    {5.2, 2.7, 3.9, 1.4, "Iris-versicolor"},
    {5.0, 2.0, 3.5, 1.0, "Iris-versicolor"},
    {5.9, 3.0, 4.2, 1.5, "Iris-versicolor"},
    {6.0, 2.2, 4.0, 1.0, "Iris-versicolor"},
    {6.1, 2.9, 4.7, 1.4, "Iris-versicolor"},
    {5.6, 2.9, 3.6, 1.3, "Iris-versicolor"},
    {6.7, 3.1, 4.4, 1.4, "Iris-versicolor"},
    {5.6, 3.0, 4.5, 1.5, "Iris-versicolor"},
    {5.8, 2.7, 4.1, 1.0, "Iris-versicolor"},
    {6.2, 2.2, 4.5, 1.5, "Iris-versicolor"},
    {5.6, 2.5, 3.9, 1.1, "Iris-versicolor"},
    {5.9, 3.2, 4.8, 1.8, "Iris-versicolor"},
    {6.1, 2.8, 4.0, 1.3, "Iris-versicolor"},
    {6.3, 2.5, 4.9, 1.5, "Iris-versicolor"},
    {6.1, 2.8, 4.7, 1.2, "Iris-versicolor"},
    {6.4, 2.9, 4.3, 1.3, "Iris-versicolor"},
    {6.6, 3.0, 4.4, 1.4, "Iris-versicolor"},
    {6.8, 2.8, 4.8, 1.4, "Iris-versicolor"},
    {6.7, 3.0, 5.0, 1.7, "Iris-versicolor"},
    {6.0, 2.9, 4.5, 1.5, "Iris-versicolor"},
    {5.7, 2.6, 3.5, 1.0, "Iris-versicolor"},
    {5.5, 2.4, 3.8, 1.1, "Iris-versicolor"},
    {5.5, 2.4, 3.7, 1.0, "Iris-versicolor"},
    {5.8, 2.7, 3.9, 1.2, "Iris-versicolor"},
    {6.0, 2.7, 5.1, 1.6, "Iris-versicolor"},
    {5.4, 3.0, 4.5, 1.5, "Iris-versicolor"},
    {6.0, 3.4, 4.5, 1.6, "Iris-versicolor"},
    {6.7, 3.1, 4.7, 1.5, "Iris-versicolor"},
    {6.3, 2.3, 4.4, 1.3, "Iris-versicolor"},
    {5.6, 3.0, 4.1, 1.3, "Iris-versicolor"},
    {5.5, 2.5, 4.0, 1.3, "Iris-versicolor"},
    {5.5, 2.6, 4.4, 1.2, "Iris-versicolor"},
    {6.1, 3.0, 4.6, 1.4, "Iris-versicolor"},
    {5.8, 2.6, 4.0, 1.2, "Iris-versicolor"},
    {5.0, 2.3, 3.3, 1.0, "Iris-versicolor"},
    {5.6, 2.7, 4.2, 1.3, "Iris-versicolor"},
    {5.7, 3.0, 4.2, 1.2, "Iris-versicolor"},
    {5.7, 2.9, 4.2, 1.3, "Iris-versicolor"},
    {6.2, 2.9, 4.3, 1.3, "Iris-versicolor"},
    {5.1, 2.5, 3.0, 1.1, "Iris-versicolor"},
    {5.7, 2.8, 4.1, 1.3, "Iris-versicolor"},
    {6.3, 3.3, 6.0, 2.5, "Iris-virginica"},
    {5.8, 2.7, 5.1, 1.9, "Iris-virginica"},
    {7.1, 3.0, 5.9, 2.1, "Iris-virginica"},
    {6.3, 2.9, 5.6, 1.8, "Iris-virginica"},
    {6.5, 3.0, 5.8, 2.2, "Iris-virginica"},
    {7.6, 3.0, 6.6, 2.1, "Iris-virginica"},
    {4.9, 2.5, 4.5, 1.7, "Iris-virginica"},
    {7.3, 2.9, 6.3, 1.8, "Iris-virginica"},
    {6.7, 2.5, 5.8, 1.8, "Iris-virginica"},
    {7.2, 3.6, 6.1, 2.5, "Iris-virginica"},
    {6.5, 3.2, 5.1, 2.0, "Iris-virginica"},
    {6.4, 2.7, 5.3, 1.9, "Iris-virginica"},
    {6.8, 3.0, 5.5, 2.1, "Iris-virginica"},
    {5.7, 2.5, 5.0, 2.0, "Iris-virginica"},
    {5.8, 2.8, 5.1, 2.4, "Iris-virginica"},
    {6.4, 3.2, 5.3, 2.3, "Iris-virginica"},
    {6.5, 3.0, 5.5, 1.8, "Iris-virginica"},
    {7.7, 3.8, 6.7, 2.2, "Iris-virginica"},
    {7.7, 2.6, 6.9, 2.3, "Iris-virginica"},
    {6.0, 2.2, 5.0, 1.5, "Iris-virginica"},
    {6.9, 3.2, 5.7, 2.3, "Iris-virginica"},
    {5.6, 2.8, 4.9, 2.0, "Iris-virginica"},
    {7.7, 2.8, 6.7, 2.0, "Iris-virginica"},
    {6.3, 2.7, 4.9, 1.8, "Iris-virginica"},
    {6.7, 3.3, 5.7, 2.1, "Iris-virginica"},
    {7.2, 3.2, 6.0, 1.8, "Iris-virginica"},
    {6.2, 2.8, 4.8, 1.8, "Iris-virginica"},
    {6.1, 3.0, 4.9, 1.8, "Iris-virginica"},
    {6.4, 2.8, 5.6, 2.1, "Iris-virginica"},
    {7.2, 3.0, 5.8, 1.6, "Iris-virginica"},
    {7.4, 2.8, 6.1, 1.9, "Iris-virginica"},
    {7.9, 3.8, 6.4, 2.0, "Iris-virginica"},
    {6.4, 2.8, 5.6, 2.2, "Iris-virginica"},
    {6.3, 2.8, 5.1, 1.5, "Iris-virginica"},
    {6.1, 2.6, 5.6, 1.4, "Iris-virginica"},
    {7.7, 3.0, 6.1, 2.3, "Iris-virginica"},
    {6.3, 3.4, 5.6, 2.4, "Iris-virginica"},
    {6.4, 3.1, 5.5, 1.8, "Iris-virginica"},
    {6.0, 3.0, 4.8, 1.8, "Iris-virginica"},
    {6.9, 3.1, 5.4, 2.1, "Iris-virginica"},
    {6.7, 3.1, 5.6, 2.4, "Iris-virginica"},
    {6.9, 3.1, 5.1, 2.3, "Iris-virginica"},
    {5.8, 2.7, 5.1, 1.9, "Iris-virginica"},
    {6.8, 3.2, 5.9, 2.3, "Iris-virginica"},
    {6.7, 3.3, 5.7, 2.5, "Iris-virginica"},
    {6.7, 3.0, 5.2, 2.3, "Iris-virginica"},
    {6.3, 2.5, 5.0, 1.9, "Iris-virginica"},
    {6.5, 3.0, 5.2, 2.0, "Iris-virginica"},
    {6.2, 3.4, 5.4, 2.3, "Iris-virginica"},
    {5.9, 3.0, 5.1, 1.8, "Iris-virginica"},
});

}  // namespace

namespace media {
namespace learning {

FisherIrisDataset::FisherIrisDataset() {
  for (auto& example : iris_examples)
    training_data_.push_back(example);
}

FisherIrisDataset::~FisherIrisDataset() = default;

const TrainingData& FisherIrisDataset::GetTrainingData() const {
  return training_data_;
}

}  // namespace learning
}  // namespace media