All Classes Namespaces Files Functions Variables Typedefs Friends Macros Pages
Kmeans.hh
Go to the documentation of this file.
1 /*
2  * Copyright (C) 2014 Open Source Robotics Foundation
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  *
16 */
17 #ifndef _IGNITION_KMEANS_HH_
18 #define _IGNITION_KMEANS_HH_
19 
20 #include <vector>
21 #include <ignition/math/Vector3.hh>
22 #include <ignition/math/Helpers.hh>
23 
24 namespace ignition
25 {
26  namespace math
27  {
34  {
37  public: Kmeans(const std::vector<Vector3d> &_obs);
38 
40  public: virtual ~Kmeans() = default;
41 
44  public: std::vector<Vector3d> Observations() const;
45 
49  public: bool Observations(const std::vector<Vector3d> &_obs);
50 
54  public: bool AppendObservations(const std::vector<Vector3d> &_obs);
55 
67  public: bool Cluster(int _k,
68  std::vector<Vector3d> &_centroids,
69  std::vector<unsigned int> &_labels);
70 
74  private: unsigned int ClosestCentroid(const Vector3d &_p) const;
75 
77  private: std::vector<Vector3d> obs;
78 
80  private: std::vector<Vector3d> centroids;
81 
83  private: std::vector<unsigned int> labels;
84 
86  private: std::vector<Vector3d> sums;
87 
89  private: std::vector<unsigned int> counters;
90  };
91  }
92 }
93 
94 #endif
#define IGNITION_VISIBLE
Use to represent "symbol visible" if supported.
Definition: Helpers.hh:346
K-Means clustering algorithm.
Definition: Kmeans.hh:33