Skip to contents

Extracts the design matrices for different parts of a heckmanGE model. The design matrices include the predictors used in the regression analysis for each component of the Generalized Heckman Model.

Usage

# S3 method for class 'heckmanGE'
model.matrix(
  object,
  part = c("selection", "outcome", "dispersion", "correlation"),
  ...
)

Arguments

object

An object of class heckmanGE. This object should be a fitted model from which the design matrices will be extracted.

part

A character vector specifying the model part for which to extract the design matrix. Options include "selection", "outcome", "dispersion", and "correlation". The default is "outcome". If multiple parts are specified, only the "outcome" part will be returned.

...

Additional arguments passed to or from other methods. These are not used in this method but must be included to match the generic method signature.

Value

A design matrix for the specified part of the heckmanGE object. If part is not one of the valid options, an error is raised.

Details

  • The function extracts the design matrix corresponding to the specified part of the heckmanGE model.

  • If the part argument is not specified correctly or includes multiple parts, the function defaults to returning the design matrix for the "outcome" part.