unweighted_plm {roseRF}R Documentation

Unweighted (baseline) estimator for the partially linear model

Description

Estimates the parameter of interest \theta_0 in the partially linear regression model

\mathbb{E}[Y|X,Z] = X\theta_0 + f_0(Z),

as in roseRF_plm but without any weights i.e. J=1, M_1(X)=X and w_1\equiv 1.

Usage

unweighted_plm(
  y_formula,
  y_learner,
  y_pars = list(),
  x_formula,
  x_learner,
  x_pars = list(),
  data,
  K = 5,
  S = 1
)

Arguments

y_formula

a two-sided formula object describing the regression model for \mathbb{E}[Y|Z].

y_learner

a string specifying the regression method to fit the regression of Y on Z as given by y_formula (e.g. randomforest, xgboost, neuralnet, gam).

y_pars

a list containing hyperparameters for the y_learner chosen. Default is an empty list, which performs hyperparameter tuning.

x_formula

a two-sided formula object describing the regression model for \mathbb{E}[X|Z].

x_learner

a string specifying the regression method to fit the regression of X on Z as given by x_formula (e.g. randomforest, xgboost, neuralnet, gam).

x_pars

a list containing hyperparameters for the y_learner chosen. Default is an empty list, which performs hyperparameter tuning.

data

a data frame containing the variables for the partially linear model.

K

the number of folds used for K-fold cross-fitting. Default is 5.

S

the number of repeats to mitigate the randomness in the estimator on the sample splits used for K-fold cross-fitting. Default is 5.

Value

A list containing:

theta

The estimator of \theta_0.

stderror

Huber robust estimate of the standard error of the \theta_0-estimator.

coefficients

Table of \theta_0 coefficient estimator, standard error, z-value and p-value.


[Package roseRF version 0.1.0 Index]