def separate_grid_model_poisson(covariates,
active='active',
regularizer_weight=None,
log_shift=.5,
log_correction='max'):
def filter_func(x):
return x[x[active]]
def pred_func(x, y):
return y * x[active]
def ign_filter_func(x):
return x[~x[active]]
def ign_pred_func(x, y):
return y * (~x[active])
afm = prg.PoissonRegressionGridModel(covariates, regularizer_weight,
log_shift, log_correction,
filter_func, pred_func)
igm = prg.PoissonRegressionGridModel(covariates, regularizer_weight,
log_shift, log_correction,
ign_filter_func, ign_pred_func)
model = aig.ActiveIgnitionGridModel(afm, igm)
return model
def active_only_grid_model_linear(covariates,
active='active',
regularizer_weight=None,
log_shift=1,
log_correction='add'):
def filter_func(x):
return x[x[active]]
def pred_func(x, y):
return y * x[active]
afm = lrg.LinearRegressionGridModel(covariates, regularizer_weight,
filter_func, pred_func)
model = aig.ActiveIgnitionGridModel(afm, None)
return model
def active_only_grid_model_poisson_hurdle(covariates,
active='active',
regularizer_weight=None,
log_shift=1,
log_correction='add'):
def filter_func(x):
return x[x[active]]
def pred_func(x, y):
return y * x[active]
afm = prhg.PoissonRegressionHurdleGridModel(covariates, regularizer_weight,
log_shift, log_correction,
filter_func, pred_func)
model = aig.ActiveIgnitionGridModel(afm, None)
return model
def active_ig_grid_model_poisson(covariates):
def filter_func(x):
return x[x[x.active]]
def pred_func(x, y):
return y * x[x.active]
afm = prg.PoissonRegressionGridModel(covariates, filter_func, pred_func)
def filter_func(x):
return x[not x.active]
def pred_func(x, y):
return y * (not x.active)
ifm = prg.PoissonRegressionGridModel(covariates, filter_func, pred_func)
model = aig.ActiveIgnitionGridModel(afm, ifm)
return model
def no_ignition_grid_model_poisson(covariates):
afm = prg.PoissonRegressionGridModel(covariates)
model = aig.ActiveIgnitionGridModel(afm, None)
return model
def only_zero_grid_model(covariates):
model = aig.ActiveIgnitionGridModel(None, None)
return model
def no_ignition_model_poisson_zip(covariates, bounding_box):
afm = gp.GridPredictorModel(
pzip.PoissonRegressionZeroInflatedModel(covariates), bounding_box)
model = aig.ActiveIgnitionGridModel(afm, None)
return model
def auto_grid_model(covariates):
igm = ag.AutoregressiveGridModel()
model = aig.ActiveIgnitionGridModel(None, igm)
return model
def only_bias_grid_model(covariates):
igm = bg.BiasGridModel()
model = aig.ActiveIgnitionGridModel(None, igm)
return model