1 | // Copyright (C) 2009 foam |
---|

2 | // |
---|

3 | // This program is free software; you can redistribute it and/or modify |
---|

4 | // it under the terms of the GNU General Public License as published by |
---|

5 | // the Free Software Foundation; either version 2 of the License, or |
---|

6 | // (at your option) any later version. |
---|

7 | // |
---|

8 | // This program is distributed in the hope that it will be useful, |
---|

9 | // but WITHOUT ANY WARRANTY; without even the implied warranty of |
---|

10 | // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
---|

11 | // GNU General Public License for more details. |
---|

12 | // |
---|

13 | // You should have received a copy of the GNU General Public License |
---|

14 | // along with this program; if not, write to the Free Software |
---|

15 | // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. |
---|

16 | |
---|

17 | #include <cfloat> |
---|

18 | #include "PCAClassifier.h" |
---|

19 | |
---|

20 | using namespace std; |
---|

21 | |
---|

22 | PCAClassifier::PCAClassifier(const PCA &pca) : |
---|

23 | Classifier(pca.GetParamsSize()), |
---|

24 | m_PCA(pca) |
---|

25 | { |
---|

26 | } |
---|

27 | |
---|

28 | PCAClassifier::~PCAClassifier() |
---|

29 | { |
---|

30 | } |
---|

31 | |
---|

32 | void PCAClassifier::AddFeature(int group, const Vector<float> &f) |
---|

33 | { |
---|

34 | Vector<float> p=m_PCA.Project(f); |
---|

35 | AddFeatureToGroup(group, p); |
---|

36 | } |
---|

37 | |
---|

38 | int PCAClassifier::Classify(const Vector<float> &f, float &error) |
---|

39 | { |
---|

40 | Vector<float> params=m_PCA.Project(f); |
---|

41 | |
---|

42 | // find the closest point in all the group means |
---|

43 | error = FLT_MAX; |
---|

44 | int ret=-1; |
---|

45 | for (map<int,Vector<float> >::iterator i=m_GroupMeans.begin(); |
---|

46 | i!=m_GroupMeans.end(); ++i) |
---|

47 | { |
---|

48 | float d=params.DistanceFrom(i->second); |
---|

49 | if (d<error) |
---|

50 | { |
---|

51 | ret=i->first; |
---|

52 | error=d; |
---|

53 | } |
---|

54 | } |
---|

55 | |
---|

56 | return ret; |
---|

57 | } |
---|

58 | |
---|

59 | void PCAClassifier::Refresh() |
---|

60 | { |
---|

61 | |
---|

62 | } |
---|

63 | |
---|