def model_higher_is_better(request):
model = LinearRegression(metric=request.param, early_stop=True,
val_size=0.3, precision=0.1,
patience=2)
model.cost_function = RegressionCostFactory()(cost='quadratic')
model.scorer = RegressionMetricFactory()(metric=request.param)
return model
def models_by_metric(request):
model = LinearRegression(metric=request.param)
model.cost_function = RegressionCostFactory()(cost='quadratic')
model.scorer = RegressionMetricFactory()(metric=request.param)
return model