registry.h

Go to the documentation of this file.

#ifndef BALL_QSAR_REGISTRY_H
#define BALL_QSAR_REGISTRY_H

#ifndef MODEL_FACTORY_H
#include <BALL/QSAR/modelFactory.h>
#endif

#include <map>

#ifdef BALL_HAS_LIBSVM
# include <BALL/QSAR/libsvmModel.h>
#endif

namespace BALL
{
  namespace QSAR
  {
    class QSARData;

    typedef Model* (*CreateMethod) (const QSARData& q);
    typedef Model* (*CreateKernel1) (const QSARData& q, int k, double p1, double p2);
    typedef Model* (*CreateKernel2) (const QSARData& q, String s1, String s2);
    
    class Registry;
    
    class BALL_EXPORT RegistryEntry
    {
      public:
      
        RegistryEntry(bool k, bool r, String n, String ab, CreateMethod c0);
        
        RegistryEntry(bool k, bool r, String n, String ab, CreateKernel1 c1, CreateKernel2 c2);
        
        RegistryEntry(const RegistryEntry& entry);
        
        ~RegistryEntry();
        
        const std::map<unsigned int,String>* getStatistics();
        String getStatName(int s);

        bool kernel;
        bool regression; // regression or classification?!
        bool latent_variables; // is this a latent variable model?!
        std::multiset<int> optimizableParameters;
        //bool optimizable;
        String name; // name of the Model
        String name_abreviation;
        CreateMethod create;
        CreateKernel1 createKernel1;
        CreateKernel2 createKernel2;
        vector<String> parameterNames;
        vector<double> parameterDefaults;
        Registry* getRegistry();
        
      private:
        Registry* registry_;
        
        friend class Registry;
    };
    
    
    typedef std::map<int,RegistryEntry>::iterator RegistryEntryIterator;
    
    class BALL_EXPORT Registry
    {
      public:
        Registry();

        ~Registry();
        
        double default_kernel_par1;
        
        double default_kernel_par2;
        
        double default_rbf_par;
        
        int default_k;
        
        int default_no_permutations;
        
        int default_no_boostrap_samples;
        
        double default_valdition_fraction;
        
        int default_nested_folds;
        
        double default_correlation_cutoff;
        
        double default_gridsearch_stepwidth;
        int default_gridsearch_recursion;
        int default_gridsearch_steps;
        double default_gridsearch_par1_start;
        double default_gridsearch_par2_start;
        
        void addEntry(RegistryEntry entry, int uniqueID);
        
        RegistryEntry* getEntry(String model_name);
        
        RegistryEntry* getEntry(int ID);
        
        int getModelNo(String model_name);
        
        String getClassificationStatisticName(unsigned int no);
        String getRegressionStatisticName(unsigned int no);
        String getFeatureSelectionName(unsigned int no);
        String getValidationName(unsigned int no);
        const std::map<unsigned int,String>* getClassificationStatistics();
        const std::map<unsigned int,String>* getRegressionStatistics();
        
        RegistryEntryIterator beginEntry();
        
        RegistryEntryIterator endEntry();
        
        
      private:
        std::map<int,RegistryEntry> registered_models;
        
        std::map<String,int> model_map;
        
        std::map<unsigned int,String> classification_statistics;
        std::map<unsigned int,String> regression_statistics;
        std::map<unsigned int,String> feature_selection_names;
        std::map<unsigned int,String> validation_names;
        
        friend class RegistryEntry;
    };
    

  }
}

#endif // BALL_QSAR_REGISTRY_H