generate_discrete_data: generate_discrete_data

In gujyjean/discretecdAlgorithm: Coordinate-Descent Algorithm for Learning Sparse Discrete Bayesian Networks

Description

data generating function

Usage

generate_discrete_data(
  graph,
  params,
  n,
  ivn = NULL,
  ivn.rand = TRUE,
  n_levels = NULL
)

Arguments

graph	A edgeList object.
params	Coefficient list.
n	Size of the data set, a scalar
ivn	List of interventions.
ivn.rand	If TRUE, random values will be drawn uniformly for each intervention. Otherwise, these values need to supplied manually in ivn.
n_levels	A vector of number of levels for each node. Default is binary data.

Value

data matrix

Examples

### generate observational data
gr <- sparsebnUtils::random.graph(5, 5) # use sparsebnUtils package to generate a random graph
names(gr) = c("V1", "V2", "V3", "V4", "V5")
nlevels <- c(3, 5, 2, 2, 3)
gr.params <- coef_gen(edge_list = gr, n_levels = nlevels)
data.obs <- discretecdAlgorithm::generate_discrete_data(graph = gr,
                                                        n = 100,
                                                        n_levels = nlevels,
                                                        params = gr.params)

### generate experimental data
ivn <- as.list(c(rep("V1", 50), rep("V2", 50))) # 50 interventions on V1, 50 interventions on V2
data.ivn <- discretecdAlgorithm::generate_discrete_data(graph = gr,
                                             n = 100,
                                             n_levels = nlevels,
                                             params = gr.params,
                                             ivn = ivn)

###  Use pre-specified values for interventions
###  In this toy example, we assume that all intervened nodes were fixed to
###  to the value 1, although this can be any number of course.
ivn.vals <- lapply(ivn, function(x) sapply(x, function(x) 1)) # replace all entries with a 1
data.ivn <- discretecdAlgorithm::generate_discrete_data(graph = gr,
                                             n = 100,
                                             n_levels = nlevels,
                                             params = gr.params,
                                             ivn = ivn.vals,
                                             ivn.rand = FALSE)

gujyjean/discretecdAlgorithm documentation built on March 15, 2020, 7:32 p.m.