mixle.models.dependence module

Conditional-dependence and small causal-structure learning utilities.

class ConditionalIndependenceResult(measure, statistic, p_value, independent)

Bases: object

Result from a conditional independence calculation.

Parameters:

• measure (float)

• statistic (float)

• p_value (float | None)

• independent (bool)

measure: float

statistic: float

p_value: float | None

independent: bool

class CausalSkeleton(edges, separating_sets, variable_names)

Bases: object

Undirected skeleton plus separating sets from a PC-style search.

Parameters:

• edges (set[tuple[int, int]])

• separating_sets (dict[tuple[int, int], frozenset[int]])

• variable_names (list[Any])

edges: set[tuple[int, int]]

separating_sets: dict[tuple[int, int], frozenset[int]]

variable_names: list[Any]

has_edge(i, j)

Return whether the undirected skeleton contains edge i–j.

Parameters:

• i (int)

• j (int)

Return type:

bool

class PartiallyDirectedGraph(directed_edges, undirected_edges, variable_names)

Bases: object

Partially directed graph after collider orientation.

Parameters:

• directed_edges (set[tuple[int, int]])

• undirected_edges (set[tuple[int, int]])

• variable_names (list[Any])

directed_edges: set[tuple[int, int]]

undirected_edges: set[tuple[int, int]]

variable_names: list[Any]

gaussian_partial_correlation(data, x, y, given=(), ridge=1.0e-10)

Return partial correlation rho_xy.given for continuous data.

Parameters:

• data (Any)

• x (int)

• y (int)

• given (Sequence[int])

• ridge (float)

Return type:

float

gaussian_conditional_independence(data, x, y, given=(), alpha=0.05, ridge=1.0e-10)

Fisher-z Gaussian conditional independence test.

Parameters:

• data (Any)

• x (int)

• y (int)

• given (Sequence[int])

• alpha (float)

• ridge (float)

Return type:

ConditionalIndependenceResult

discrete_conditional_mutual_information(data, x, y, given=())

Estimate I(X;Y | Z) from categorical samples using empirical counts.

Parameters:

• data (Any)

• x (int)

• y (int)

• given (Sequence[int])

Return type:

float

learn_pc_skeleton(data, variable_names=None, alpha=0.05, max_cond_set=2, method='gaussian')

Learn a PC-style undirected skeleton from conditional independences.

Parameters:

• data (Any)

• variable_names (Sequence[Any] | None)

• alpha (float)

• max_cond_set (int)

• method (str)

Return type:

CausalSkeleton

orient_v_structures(skeleton)

Orient unshielded colliders i -> k <- j using separating sets.

Parameters:

skeleton (CausalSkeleton)

Return type:

PartiallyDirectedGraph