include/neuralnet.h

Go to the documentation of this file.

/********************************************************************************
 *  Neural Network Framework.                                                   *
 *  Copyright (C) 2005-2008 Gianluca Massera <emmegian@yahoo.it>                *
 *                                                                              *
 *  This program is free software; you can redistribute it and/or modify        *
 *  it under the terms of the GNU General Public License as published by        *
 *  the Free Software Foundation; either version 2 of the License, or           *
 *  (at your option) any later version.                                         *
 *                                                                              *
 *  This program 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 General Public License for more details.                                *
 *                                                                              *
 *  You should have received a copy of the GNU General Public License           *
 *  along with this program; if not, write to the Free Software                 *
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA  *
 ********************************************************************************/

#ifndef NEURALNET_H
#define NEURALNET_H

#include "types.h"
#include "clonable.h"
#include "cluster.h"
#include "linker.h"
#include <map>
#include <string>

namespace nnfw {

class NNFW_API BaseNeuralNet : public Clonable {
public:

    BaseNeuralNet();

    ~BaseNeuralNet();



    void addCluster( Cluster* c, bool isInput = false, bool isOutput = false );

    void addInputCluster( Cluster* c ) {
        addCluster( c, true, false );
    };

    void addOutputCluster( Cluster* c ) {
        addCluster( c, false, true );
    };

    bool removeCluster( Cluster* c );

    void markAsInput( Cluster* c );

    void markAsOutput( Cluster* c );

    void unmark( Cluster* c );

    void unmarkAll();

    bool isIsolated( Cluster* c ) const;

    const ClusterVec& clusters() const;

    const ClusterVec& inputClusters() const;

    const ClusterVec& outputClusters() const;

    const ClusterVec& hiddenClusters() const;

    void addLinker( Linker* l );

    bool removeLinker( Linker* );

    const LinkerVec& linkers() const;

    const LinkerVec& linkers( Cluster* c, bool out = false ) const;

    void setOrder( Updatable* updatables[], u_int dim );

    void setOrder( const UpdatableVec& );

    const UpdatableVec& order() const {
        return ups;
    };

    void step() {
        for( u_int i=0; i<dimUps; i++ ) {
            ups[i]->update();
        }
    };

    void randomize( Real min, Real max );

    template<class PointerTo>
    PointerTo byName( const char* aName, PointerTo& aPointer ) {
        aPointer = dynamic_cast<PointerTo>( getByName(aName) );
        return aPointer;
    };

    Updatable* getByName( const char* );

    bool find( const Cluster* ) const;

    bool find( const Linker* ) const;

    bool find( const Updatable* ) const;

    BaseNeuralNet* clone() const;


protected:
    ClusterVec clustersv;
    ClusterVec inclusters;
    ClusterVec outclusters;
    ClusterVec hidclusters;
    LinkerVec  linkersv;

    typedef std::map<std::string, Cluster*> ClustersMap;
    ClustersMap clsMap;
    class ids4t {
    public:
        int& operator[]( int i ) { return ids[i]; };
        int ids[4];
    };
    //typedef int ids4t [4];
    typedef std::map<Cluster*, ids4t> IdsMap;
    IdsMap clsIdsMap;

    typedef std::map<Cluster*, LinkerVec> LinkVecMap;
    LinkVecMap inLinks;
    LinkVecMap outLinks;

    typedef std::map<std::string, Linker*> LinkersMap;
    LinkersMap lksMap;
    typedef std::map<Linker*, ids4t> IdsMapL;
    IdsMapL lksIdsMap;

    UpdatableVec ups;
    unsigned int dimUps;
};

}

#endif

Copyright © 2005-2008 G. Massera

Contact: emmegian (at) yahoo (dot) it

  Documentation generated by