nnfw/include/matrixlinker.h

Go to the documentation of this file.

/********************************************************************************
 *  Neural Network Framework.                                                   *
 *  Copyright (C) 2005-2011 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 MATRIXLINKER_H
#define MATRIXLINKER_H

#include "nnfwconfig.h"
#include "linker.h"

namespace farsa {

class FARSA_NNFW_API MatrixLinker : public Linker {
public:
    MatrixLinker( Cluster* from, Cluster* to, QString name = "unnamed" );
    MatrixLinker( ConfigurationParameters& params, QString prefix );
    virtual ~MatrixLinker();
    unsigned int rows() {
        return nrows;
    };
    unsigned int cols() {
        return ncols;
    };
    unsigned int size() const;
    virtual void randomize( double min, double max );
    virtual void setWeight( unsigned int from, unsigned int to, double weight );
    virtual double weight( unsigned int from, unsigned int to );
    DoubleMatrix& matrix() {
        return w;
    };
    DoubleMatrix matrix() const {
        return w;
    };
    void connect( unsigned int from, unsigned int to );
    void connectRandom( double prob, bool zeroDiagonal = false, bool symmetric = false );
    void connectAll();
    void disconnect( unsigned int from, unsigned int to );
    void disconnectAll();
    virtual void save(ConfigurationParameters& params, QString prefix);
private:
    unsigned int nrows, ncols;
    DoubleMatrix w;
};

}

#endif