def __init__(self):
     self._name = 'LearningCurves'
     AnalysisModule.__init__(self)
     self.models = [LinearModel, RidgeModel, GBMModel, SVMLinearModel]
     self.model_names = [
         str(self.models[i]()) for i in range(0, len(self.models))
     ]
     self.model_colors = {
         str(model()): model()._color
         for model in self.models
     }
     self.nreps = range(1, 11)
Example #2
0
 def __init__(self):
     self._name = 'PredictionRmse'
     AnalysisModule.__init__(self)
Example #3
0
 def __init__(self):
     self._name = 'ApplicationTimes'
     AnalysisModule.__init__(self)
Example #4
0
 def __init__(self):
     self._name = 'FeatureSelection'
     self._models = [LinearModel, RidgeModel, GBMModel, SVMLinearModel]
     self._cost_function = util.rmse_error
     AnalysisModule.__init__(self)
Example #5
0
 def __init__(self):
     self._name = 'TrainingSetSize'
     AnalysisModule.__init__(self)
     self.models = [LinearModel, RidgeModel, GBMModel, SVMLinearModel]