BALL  1.5.0
registry.h
Go to the documentation of this file.
1 #ifndef BALL_QSAR_REGISTRY_H
2 #define BALL_QSAR_REGISTRY_H
3 
4 #ifndef MODEL_FACTORY_H
6 #endif
7 
8 #include <map>
9 
10 #ifdef BALL_HAS_LIBSVM
11 # include <BALL/QSAR/libsvmModel.h>
12 #endif
13 
14 namespace BALL
15 {
16  namespace QSAR
17  {
18  class QSARData;
19 
20  typedef Model* (*CreateMethod) (const QSARData& q);
21  typedef Model* (*CreateKernel1) (const QSARData& q, int k, double p1, double p2);
22  typedef Model* (*CreateKernel2) (const QSARData& q, String s1, String s2);
23 
24  class Registry;
25 
27  {
28  public:
29 
34  RegistryEntry(bool k, bool r, String n, String ab, CreateMethod c0);
35 
40  RegistryEntry(bool k, bool r, String n, String ab, CreateKernel1 c1, CreateKernel2 c2);
41 
43 
45 
46  const std::map<unsigned int,String>* getStatistics();
48 
49  bool kernel;
50  bool regression; // regression or classification?!
51  bool latent_variables; // is this a latent variable model?!
52  std::multiset<int> optimizableParameters;
53  //bool optimizable;
54  String name; // name of the Model
59  vector<String> parameterNames;
60  vector<double> parameterDefaults;
62 
63  private:
64  Registry* registry_;
65 
66  friend class Registry;
67  };
68 
69 
70  typedef std::map<int,RegistryEntry>::iterator RegistryEntryIterator;
71 
73  {
74  public:
76 
78 
81 
84 
87 
89  int default_k;
90 
93 
96 
99 
102 
105 
111 
112  void addEntry(RegistryEntry entry, int uniqueID);
113 
116 
119 
121  int getModelNo(String model_name);
122 
125  String getFeatureSelectionName(unsigned int no);
126  String getValidationName(unsigned int no);
127  const std::map<unsigned int,String>* getClassificationStatistics();
128  const std::map<unsigned int,String>* getRegressionStatistics();
129 
132 
135 
136 
137  private:
138  std::map<int,RegistryEntry> registered_models;
139 
141  std::map<String,int> model_map;
142 
143  std::map<unsigned int,String> classification_statistics;
144  std::map<unsigned int,String> regression_statistics;
145  std::map<unsigned int,String> feature_selection_names;
146  std::map<unsigned int,String> validation_names;
147 
148  friend class RegistryEntry;
149  };
150 
151 
152  }
153 }
154 
155 #endif // BALL_QSAR_REGISTRY_H
BALL::QSAR::Registry::endEntry
RegistryEntryIterator endEntry()
libsvmModel.h
BALL_EXPORT
#define BALL_EXPORT
Definition: COMMON/global.h:50
BALL::QSAR::RegistryEntry::RegistryEntry
RegistryEntry(bool k, bool r, String n, String ab, CreateMethod c0)
BALL::QSAR::Registry::getModelNo
int getModelNo(String model_name)
BALL::QSAR::Registry::getEntry
RegistryEntry * getEntry(int ID)
BALL::QSAR::RegistryEntry::getStatistics
const std::map< unsigned int, String > * getStatistics()
BALL::QSAR::RegistryEntry::parameterDefaults
vector< double > parameterDefaults
Definition: registry.h:60
BALL::QSAR::CreateKernel1
Model *(* CreateKernel1)(const QSARData &q, int k, double p1, double p2)
Definition: registry.h:21
BALL::QSAR::Registry::getClassificationStatisticName
String getClassificationStatisticName(unsigned int no)
BALL::QSAR::Registry::default_valdition_fraction
double default_valdition_fraction
Definition: registry.h:98
BALL::QSAR::Registry::Registry
Registry()
BALL::QSAR::Registry::default_correlation_cutoff
double default_correlation_cutoff
Definition: registry.h:104
BALL::QSAR::Registry::beginEntry
RegistryEntryIterator beginEntry()
BALL::QSAR::RegistryEntry::regression
bool regression
Definition: registry.h:50
BALL::QSAR::RegistryEntry::name
String name
Definition: registry.h:54
BALL::QSAR::Registry::getFeatureSelectionName
String getFeatureSelectionName(unsigned int no)
BALL::QSAR::RegistryEntry::latent_variables
bool latent_variables
Definition: registry.h:51
BALL::QSAR::Registry
Definition: registry.h:73
BALL::QSAR::RegistryEntry::optimizableParameters
std::multiset< int > optimizableParameters
Definition: registry.h:52
BALL::String
Definition: string.h:57
BALL::QSAR::Registry::default_rbf_par
double default_rbf_par
Definition: registry.h:86
BALL
Definition: constants.h:13
BALL::QSAR::QSARData
Definition: QSARData.h:56
BALL::QSAR::Registry::default_gridsearch_recursion
int default_gridsearch_recursion
Definition: registry.h:107
BALL::QSAR::Registry::getEntry
RegistryEntry * getEntry(String model_name)
BALL::QSAR::Registry::getRegressionStatistics
const std::map< unsigned int, String > * getRegressionStatistics()
BALL::QSAR::RegistryEntry::RegistryEntry
RegistryEntry(const RegistryEntry &entry)
BALL::QSAR::RegistryEntry::parameterNames
vector< String > parameterNames
Definition: registry.h:59
BALL::QSAR::Registry::getRegressionStatisticName
String getRegressionStatisticName(unsigned int no)
BALL::QSAR::Registry::getClassificationStatistics
const std::map< unsigned int, String > * getClassificationStatistics()
BALL::QSAR::RegistryEntry::kernel
bool kernel
Definition: registry.h:49
BALL::QSAR::Registry::default_no_boostrap_samples
int default_no_boostrap_samples
Definition: registry.h:95
BALL::QSAR::Registry::default_gridsearch_steps
int default_gridsearch_steps
Definition: registry.h:108
BALL::QSAR::CreateMethod
Model *(* CreateMethod)(const QSARData &q)
Definition: registry.h:20
BALL::QSAR::RegistryEntry::getStatName
String getStatName(int s)
BALL::QSAR::RegistryEntry::~RegistryEntry
~RegistryEntry()
BALL::QSAR::RegistryEntry::createKernel2
CreateKernel2 createKernel2
Definition: registry.h:58
BALL::QSAR::Registry::getValidationName
String getValidationName(unsigned int no)
BALL::QSAR::Registry::addEntry
void addEntry(RegistryEntry entry, int uniqueID)
BALL::QSAR::Registry::default_gridsearch_par2_start
double default_gridsearch_par2_start
Definition: registry.h:110
BALL::QSAR::Registry::default_nested_folds
int default_nested_folds
Definition: registry.h:101
BALL::QSAR::Registry::default_gridsearch_stepwidth
double default_gridsearch_stepwidth
Definition: registry.h:106
BALL::QSAR::Registry::default_gridsearch_par1_start
double default_gridsearch_par1_start
Definition: registry.h:109
BALL::QSAR::Registry::default_no_permutations
int default_no_permutations
Definition: registry.h:92
BALL::QSAR::Registry::default_kernel_par1
double default_kernel_par1
Definition: registry.h:80
BALL::QSAR::Registry::default_kernel_par2
double default_kernel_par2
Definition: registry.h:83
BALL::QSAR::Registry::~Registry
~Registry()
BALL::QSAR::CreateKernel2
Model *(* CreateKernel2)(const QSARData &q, String s1, String s2)
Definition: registry.h:22
BALL::QSAR::RegistryEntry::getRegistry
Registry * getRegistry()
BALL::Constants::k
BALL_EXTERN_VARIABLE const double k
Definition: constants.h:93
BALL::QSAR::Model
Definition: Model.h:35
BALL::QSAR::RegistryEntry
Definition: registry.h:27
BALL::QSAR::RegistryEntry::createKernel1
CreateKernel1 createKernel1
Definition: registry.h:57
BALL::QSAR::RegistryEntryIterator
std::map< int, RegistryEntry >::iterator RegistryEntryIterator
Definition: registry.h:70
BALL::QSAR::RegistryEntry::RegistryEntry
RegistryEntry(bool k, bool r, String n, String ab, CreateKernel1 c1, CreateKernel2 c2)
BALL::QSAR::RegistryEntry::name_abreviation
String name_abreviation
Definition: registry.h:55
BALL::QSAR::Registry::default_k
int default_k
Definition: registry.h:89
BALL::QSAR::RegistryEntry::create
CreateMethod create
Definition: registry.h:56
modelFactory.h