ps_svm: Fitting SVM for Propensity Score

In ygeunkim/propensityml: Propensity score weighting using machine learning methods

Description

Fits Support Vector Machine before estimating propensity scores

Usage

ps_svm(
  formula,
  data,
  scale = FALSE,
  kernel = c("radial", "linear", "polynomial", "sigmoid"),
  cost = 1,
  gamma,
  ...
)

Arguments

formula	an object formula to be fitted. Response should be treatment.
data	optional data frame.
scale	scale the data?
kernel	kernel used in SVM. See svm
cost	cost parameter C
gamma	parameter in some kernel
...	For additional options of svm

Value

propmod class, a list with model and its name

• model - svm

• name - "SVM"

• data - dataset

ygeunkim/propensityml documentation built on Jan. 1, 2021, 1:44 p.m.