fitModel.h

Go to the documentation of this file.

/* fitModel.h
 * 
 * Copyright (C) 2009 Marcel Schumann
 * 
 * This file is part of QuEasy -- A Toolbox for Automated QSAR Model
 * Construction and Validation.
 * QuEasy 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 3 of the License, or (at
 * your option) any later version.
 * 
 * QuEasy 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, see <http://www.gnu.org/licenses/>.
 */

// -*- Mode: C++; tab-width: 2; -*-
// vi: set ts=2:
//
//

#ifndef FMODEL
#define FMODEL

#ifndef NLMODEL
#include <BALL/QSAR/nonlinearModel.h>
#endif

#include <gsl/gsl_multifit_nlin.h>
#include <gsl/gsl_deriv.h>
#include <gsl/gsl_math.h>
#include <BALL/MATHS/parsedFunction.h>

//#include <list>

namespace BALL
{
  namespace QSAR
  {
    class BALL_EXPORT FitModel : public NonLinearModel
    { 
      public:
        FitModel(const QSARData& q);
        
        FitModel(const QSARData& q, vector<String>& eq, vector<double>& guess);
        
        FitModel(const QSARData& q, vector<String>& eq, vector<vector<String> >& deq, vector<double>& guess);

        //~FitModel();
        
        
        void setEquations(vector<String>& eq, vector<vector<String> >& deq);
        
        void setEquations(vector<String>& eq);
      
        void train();
        
        Vector<double> predict(const vector<double>& substance, bool transform=1);
        
        void setInitialGuess(vector<double>& d);
        
        
      private:
        vector<String> allEquations_;
        
        vector<vector<String> > allDiffEquations_;
        
        //vector<String> allCoefficients;
        
        vector<double> initial_guess_;

    };
  

        
    const Matrix<double>* fitX;
    Matrix<double>* fitY;
    String* equation;
    vector<String>* diffEquations;
    ParsedFunction<double>* f;
    unsigned int c; // current column
    
    double getFunctionValue(double X, void* params);
        
    int setF(const gsl_vector* x, void* params, gsl_vector* f);
      
    int setDf(const gsl_vector * x, void* params, gsl_matrix * df);
    
    int setFdf (const gsl_vector * x, void *params, gsl_vector * f, gsl_matrix * df);
    
    gsl_multifit_function_fdf make_fdf (int (* f) (const gsl_vector *, void *, gsl_vector *),
        int (* df) (const gsl_vector *, void *, gsl_matrix *),
        int (* fdf) (const gsl_vector *, void *, gsl_vector *, gsl_matrix *),
        size_t n,
        size_t p, void * params );
    
  }
}

#endif // FMODEL