1import numpy as np
2from causallearn.graph.GeneralGraph import GeneralGraph
3from causallearn.graph.GraphNode import GraphNode
4
5# =====================================================================
6# EDITABLE: implement run_causal_discovery below
7# =====================================================================
8def run_causal_discovery(X: np.ndarray) -> GeneralGraph:
9    """
10    Input:  X of shape (n_samples, n_variables), integer-encoded discrete data
11    Output: estimated CPDAG as causallearn.graph.GeneralGraph.GeneralGraph
12    """
13    nodes = [GraphNode(f"X{i + 1}") for i in range(X.shape[1])]
14    return GeneralGraph(nodes)
15# =====================================================================
... 1 more lines

1"""Evaluation harness for the causal-discovery-discrete task."""
2import argparse
3import os
4import sys
5
6sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
7
8from custom_algorithm import run_causal_discovery
9from data_gen import load_and_sample
10from metrics import compute_metrics
11
12
13def main():
14    parser = argparse.ArgumentParser(
15        description="Evaluate CPDAG recovery on bnlearn discrete data."
... 46 more lines

1"""Data loader for bnlearn Bayesian network benchmarks.
2
3Uses pgmpy's bundled BIF files (no network access needed) to load
4real-world Bayesian networks, sample discrete observational data,
5and extract the ground-truth DAG.
6"""
7import numpy as np
8import pandas as pd
9
10from causallearn.graph.Dag import Dag
11from causallearn.graph.GraphNode import GraphNode
12
13# All discrete bnlearn networks supported by pgmpy
14SUPPORTED_NETWORKS = [
15    "cancer", "earthquake", "survey", "asia", "sachs",
... 58 more lines

1"""Evaluation metrics for CPDAG recovery on bnlearn discrete data."""
2from causallearn.graph.AdjacencyConfusion import AdjacencyConfusion
3from causallearn.graph.ArrowConfusion import ArrowConfusion
4from causallearn.graph.Graph import Graph
5from causallearn.graph.SHD import SHD
6from causallearn.utils.DAG2CPDAG import dag2cpdag
7
8
9def _safe_div(numerator, denominator):
10    return numerator / denominator if denominator > 0 else 0.0
11
12
13def _normalize_graph_output(graph_output):
14    """Normalize algorithm output to a causallearn Graph object."""
15    if isinstance(graph_output, dict) and "G" in graph_output:
... 41 more lines