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)
def __init__(self):
self._name = 'PredictionRmse'
AnalysisModule.__init__(self)
def __init__(self):
self._name = 'ApplicationTimes'
AnalysisModule.__init__(self)
def __init__(self):
self._name = 'FeatureSelection'
self._models = [LinearModel, RidgeModel, GBMModel, SVMLinearModel]
self._cost_function = util.rmse_error
AnalysisModule.__init__(self)
def __init__(self):
self._name = 'TrainingSetSize'
AnalysisModule.__init__(self)
self.models = [LinearModel, RidgeModel, GBMModel, SVMLinearModel]