causalnex.structure

Description

causalnex.structure provides functionality to define or learn structure.

Modules

causalnex.structure.notears Tools to learn a StructureModel which describes the conditional dependencies between variables in a dataset.
causalnex.structure.dynotears Tools to learn a Dynamic Bayesian Network which describe the conditional dependencies between variables in a time-series dataset.
causalnex.structure.data_generators Data generators using DAGs for benchmarking and synthetic data generation.

Classes

DAGClassifier([dist_type_schema, alpha, …]) Classifier wrapper of the StructureModel.
DAGRegressor([dist_type_schema, alpha, …]) Regressor wrapper of the StructureModel.
StructureModel([incoming_graph_data, origin]) Base class for structure models, which are an extension of networkx.DiGraph.