Skip to content

X_train and X_test are not consistently processed for bart propensity model #314

@MaggieWang0614

Description

@MaggieWang0614

stochtree/R/bcf.R

Line 1196 in 7e2e714

X_train = X_train,

X_train is Preprocessed on line 912, but X_test_raw is not preprocessed for bart model. This would raise the following error:

Error in [.data.frame(input_df, , numeric_vars, drop = FALSE): undefined columns selected
Traceback:

  1. bart(X_train = X_train, y_train = as.numeric(Z_train), X_test = X_test_raw,
    . num_gfr = num_total, num_burnin = 0, num_mcmc = 0)
  2. preprocessPredictionData(X_test, X_train_metadata)
  3. preprocessPredictionDataFrame(input_data, metadata)
  4. input_df[, numeric_vars, drop = FALSE]
  5. [.data.frame(input_df, , numeric_vars, drop = FALSE)
  6. stop("undefined columns selected")
  7. .handleSimpleError(function (cnd)

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions