gfn.gym.helpers.bayesian_structure.sampling

Functions

sample_from_linear_gaussian(graph, num_samples[, rng])	Sample from a linear-Gaussian model using ancestral sampling.
topological_sort_data(graph)

Module Contents

gfn.gym.helpers.bayesian_structure.sampling.sample_from_linear_gaussian(graph, num_samples, rng=np.random.default_rng())

Sample from a linear-Gaussian model using ancestral sampling.

Parameters:

• graph (torch_geometric.data.Data)

• num_samples (int)

• rng (numpy.random.Generator)

Return type:

pandas.DataFrame

gfn.gym.helpers.bayesian_structure.sampling.topological_sort_data(graph)

Parameters:

graph (torch_geometric.data.Data)

Return type:

list[int]