vtreat (version 1.6.4)

NumericOutcomeTreatment: Stateful object for designing and applying numeric outcome treatments.

Description

Hold settings and results for regression data preparation.

Usage

NumericOutcomeTreatment(
  ...,
  var_list,
  outcome_name,
  cols_to_copy = NULL,
  params = NULL,
  imputation_map = NULL
)

Arguments

...

not used, force arguments to be specified by name.

var_list

Names of columns to treat (effective variables).

outcome_name

Name of column holding outcome variable. dframe[[outcomename]] must be only finite non-missing values.

cols_to_copy

list of extra columns to copy.

params

parameters list from regression_parameters

imputation_map

map from column names to functions of signature f(values: numeric, weights: numeric), simple missing value imputers.

Details

Please see https://github.com/WinVector/vtreat/blob/main/Examples/fit_transform/fit_transform_api.md, mkCrossFrameNExperiment, designTreatmentsN, and prepare.treatmentplan for details.