File: clustererFlame.h

package info (click to toggle)
mldemos 0.5.1-3
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 32,224 kB
  • ctags: 46,525
  • sloc: cpp: 306,887; ansic: 167,718; ml: 126; sh: 109; makefile: 2
file content (110 lines) | stat: -rw-r--r-- 3,440 bytes parent folder | download | duplicates (2)
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
/*********************************************************************
FLAME Implementation in MLDemos
Copyright (C) Pierre-Antoine Sondag (pasondag@gmail.com) 2012

Based on the standard implementation of FLAME data clustering algorithm.
Copyright (C) 2007, Fu Limin (phoolimin@gmail.com).
All rights reserved.

MLDemos: A User-Friendly visualization toolkit for machine learning
Copyright (C) 2010  Basilio Noris
Contact: mldemos@b4silio.com

This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public License,
version 3 as published by the Free Software Foundation.

This library is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the Free
Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*********************************************************************/
#ifndef _CLUSTERER_FLAME_H_
#define _CLUSTERER_FLAME_H_

#include <boost/functional/hash.hpp>
#include <clusterer.h>
#include <flame.h>
#include <boost/unordered_map.hpp>
//#include <unordered_map>
#include <vector>

// generic hashing function
template <typename Container>
struct container_hash {
    std::size_t operator()(Container const& c) const {
        return boost::hash_range(c.begin(), c.end());
    }
};


class ClustererFlame : public Clusterer {
private:

    float **data;
    Flame *flameStructure;

    int knnParameter; // used by Flame_DefineSupports
    int knnMetric; //passed to Flame_SetDataMatrix

    int maxIterationsParameter; //used by Flame_MakeCluster
    float epsilonParameter; //used by Flame_MakeCluster

    bool isSeveralClasses; //used by Flame_MakeClusters
    float thresholdParameter;  //used by Flame_MakeClusters

    boost::unordered_map<fvec, vector<int>, container_hash<fvec> > resultMap;
    vector<fvec> supports;

    void PrintDone();


public:

    /** Constructor, instanciating everything that will be used */
    ClustererFlame(){
        data = NULL;
        flameStructure = NULL;
        knnParameter = 10;
        maxIterationsParameter = 100;
        epsilonParameter = 1e-6;
    }

    /** Deconstructor, deinstanciating everything that has been instanciated */
    ~ClustererFlame(){ }

    /** The training function, called by the main program, all training is here. */
    void Train(std::vector< fvec > samples);

    /**
    The testing function, returns a vector of size nbClusters,
    with the contribution/weight of the point for each cluster */
    fvec Test(const fvec &sample);

    /**
    Information string for the Algorithm Information and Statistics
    panel in the main program interface. Here you probably will put
    the number of parameters, the training time or anything else
    */
    const char *GetInfoString();

    /**
    Function to set the algorithm hyper-parameters, called prior to the
    training itself
    */
    void SetParams(
        int knnParameter,
        int knnMetric,
        int maxIterationsParameter,
        bool isSeveralClasses,
        float thresholdParameter);

    /* Once the treining is done, returns the vectors choosen to be supports. */
    vector<fvec> GetSupports();
};

#endif