predict.truncmodel: Predicted Expected Response for One-Inflated or Truncated...

View source: R/predict-truncmodel.R

predict.truncmodelR Documentation

Predicted Expected Response for One-Inflated or Truncated Models

Description

Calculates the predicted expected response for a model fitted using oneinfl or truncreg.

Usage

## S3 method for class 'truncmodel'
predict(object, ...)

Arguments

object

An object of class truncmodel

...

Additional argument df, a data frame used to calculate the expected value of the response variable.

Details

This function computes the expected response based on the fitted model. The computation differs depending on the distribution. For Poisson (PP), predicted values are computed using E_pois_noinfl. For Negative Binomial (ZTNB), predicted values are computed using E_negbin_noinfl.

Value

A numeric vector of predicted expected responses for the observations in df.

See Also

oneinfl for fitting one-inflated models. truncreg for fitting truncated models. E_pois_noinfl, E_negbin_noinfl for the expected value calculations.

Examples

# Example usage
df <- data.frame(x = rnorm(100), y = rpois(100, lambda = 5))
model <- truncreg(y ~ x, df = df, dist = "Poisson")
predict(model, df = df)


oneinfl documentation built on April 4, 2025, 12:05 a.m.