statsmodels.discrete.count_model.GenericZeroInflated.predict

GenericZeroInflated.predict(params, exog=None, exog_infl=None, exposure=None, offset=None, which='mean')

Predict response variable of a count model given exogenous variables.

Parameters

paramsarray_like

The parameters of the model

exogndarray, optional

A reference to the exogenous design. If not assigned, will be used exog from fitting.

exog_inflndarray, optional

A reference to the zero-inflated exogenous design. If not assigned, will be used exog from fitting.

offsetndarray, optional

Offset is added to the linear prediction with coefficient equal to 1.

exposurendarray, optional

Log(exposure) is added to the linear prediction with coefficient equal to 1. If exposure is specified, then it will be logged by the method. The user does not need to log it first.

whichstr, optional

Define values that will be predicted. ‘mean’, ‘mean-main’, ‘linear’, ‘mean-nonzero’, ‘prob-zero, ‘prob’, ‘prob-main’ Default is ‘mean’.