Learn R Programming

FLAME (version 1.0.0)

FLAME_SQLite: SQLite Database Implementation

Description

FLAME_SQLite applies the FLAME algorithm based on SQLite. FLAME_SQLite does not require external database installment. However, user should connect to a temporary database with command dbConnect(SQLite(),"tempdb_name") and name the connection as db. The required arguments include (1) db, (2) data, and (3) holdout. The rest of the arguments are optional.

Usage

FLAME_SQLite(db, data, holdout, compute_var = FALSE, tradeoff = 0.1,
  PE_function = NULL, model = NULL, ridge_reg = NULL,
  lasso_reg = NULL, tree_depth = NULL)

Arguments

db

name of the connection to temporary database (must name the connection as db)

data

input data

holdout

holdout training data

compute_var

indicator variable of computing variance (optional, default = FALSE)

tradeoff

tradeoff parameter to compute Match Quality (optional, default = 0.1)

PE_function

user defined function to compute predictive error (optional)

model

user defined model - Linear, Ridge, Lasso, or DecisionTree (optional)

ridge_reg

L2 regularization parameter if model = Ridge (optional)

lasso_reg

L1 regularization parameter if model = Lasso (optional)

tree_depth

maximum depth of decision tree if model = DecisionTree (optional)

Value

(1) list of covariates FLAME performs matching at each iteration, (2) Sizes, conditional average treatment effects (CATEs), and variance (if compute_var = TRUE) of matches at each iteration, (3) match quality at each iteration, and (4) the original data with additional column *matched*, indicating the number of covariates each unit is matched on. If a unit is never matched, then *matched* will be 0.

Examples

Run this code
# NOT RUN {
data <- data(toy_data)

db <- dbConnect(SQLite(),"tempdb")

FLAME_SQLite(db = db, data = data, holdout = data)

dbDisconnect(db)
# }

Run the code above in your browser using DataLab