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]