oplsModel.h

Go to the documentation of this file.

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

#ifndef OPLSMODEL
#define OPLSMODEL

#ifndef PLSMODEL
#include <BALL/QSAR/plsModel.h>
#endif



namespace BALL
{
  namespace QSAR
  {
    class BALL_EXPORT OPLSModel : public PLSModel
    {
      public:
        OPLSModel(const QSARData& q);

        ~OPLSModel();
        EIGEN_MAKE_ALIGNED_OPERATOR_NEW
        
        


        void train();
        
        const Eigen::MatrixXd* getWOrtho();
        
        const Eigen::MatrixXd* getTOrtho();
        
        int getNoOrthoComponents();
        
        void setNoOrthoComponents(int d);
        
        bool optimizeParameters(int k, int no_steps);
        //RowVector predict(const vector<double>& substance, bool transform);
        
        void setParameters(vector<double>& v);
        
        vector<double> getParameters() const;
        
        
      private:
        Eigen::MatrixXd T_ortho_;
        Eigen::MatrixXd W_ortho_;
        
        int no_ortho_components_;
    };
  }
}


#endif // OPLSMODEL