开源仓库pgmpy/pgmpy3,295 Stars

pgmpy

用图结构编码变量间的概率依赖和因果假设,做条件推断、因果发现与合成数据生成。

适合需要从数据中学变量关系图结构、或基于已知因果图做干预推断的研究者。从定义网络和条件概率表开始,逐步完成推理、验证和模拟。

bo老师判断:推荐

用途和边界相对清楚,可以先从小样本开始。

...互动状态已更新

这个工具提供了哪些能力

点开任一能力,看清怎么用

每个能力展开后是三件事:它会给你什么、你需要准备什么、哪些必须由你把关。

常见场景:需要从观测数据中学变量间的因果或依赖关系图;已有因果图,想知道干预某个变量后另一个变量的分布变化;需要批量生成符合已知联合分布的合成数据做方法验证;需要对比两个因果图结构与真实图的差距

覆盖阶段设计分析验证

第一次这样开始

把仓库地址直接交给 AI

先让 AI 说明环境和风险,再完成最小安装验证。开始前想清楚要用哪个能力、把对应材料备在手边——每张能力卡展开后有具体清单。

第一次直接复制

帮我安装这个库:https://github.com/pgmpy/pgmpy
安装前先告诉我需要什么环境,安装后帮我跑通一个最小示例。

别这样用

这些判断,仍然由你负责

  • 把观测相关当因果效应——只有do操作才能区分两者
  • 把学到的图结构直接当真因果图——PC等方法学到的是等价类,部分方向不可识别
  • 用离散独立性检验处理连续数据——需匹配数据类型选chi_square或fisher_z
  • check_model通过就认为图方向正确——它只校验图和CPD的对应关系

技术依据

查看能力拆解、验证范围与来源

处理机制

  1. 01

    核心处理链

    用图结构编码变量间的概率依赖和因果假设,做条件推断、因果发现与合成数据生成。

已核对范围

  • 读取 pgmpy/estimators/MLE.py(MaximumLikelihoodEstimator getparameters/estimatecpd 签名)
  • 读取 pgmpy/estimators/BayesianEstimator.py(BayesianEstimator getparameters/estimatecpd 签名)
  • 读取 pgmpy/estimators/EM.py(ExpectationMaximization)
  • 读取 pgmpy/parameterestimator/(discretebayesian/discretemle/discreteem/lineargaussianmle)
  • 读取 DiscreteBayesianNetwork.py:592 fit / :653 fitupdate 签名
  • 读取 LinearGaussianBayesianNetwork.py:880 fit 签名
  • 读取 README Parameter Learning 特性与 example notebooks 主题
  • 读取 pgmpy/inference/CausalInference.py(CausalInference 类、query(query, do, evidence, adjustmentset, inferencealgo) 签名)

仍待核对

  • 未运行本机参数学习 smoke
  • 未验证 BayesianEstimator 各先验类型(BDeu/Dirichlet/K2)的数值差异
  • 未覆盖 SEMEstimator/IVEstimator 的 SEM 参数估计内部机制(见待复核卡)
  • 未运行本机因果查询 smoke
  • 未验证 minimal/all/minimalvariance 三种调整集变体的数值差异
  • 未覆盖 DoubleMLRegressor/NaiveAdjustmentRegressor/NaiveIVRegressor(因果效应估计的预测回归器,见待复核卡)
  • 未运行本机模拟 smoke
  • 未验证 missingprob 的 MAR/MNAR 缺失机制实现细节

原仓库

pgmpy/pgmpy

读取版本

c7fa866a8b50a7eb781124e12411708dd86f2c25

许可

MIT

最近核对

内容 2026-07-10
Stars 2026-07-11

已完成源码核对

科研之我见

AI 可以参与科研的哪些环节?一张 11 环节工作流全景图

从选题、文献、理论、数据到投稿,逐环节判断 AI 可以主导、需要人工校验和必须由研究者把关的工作。

查看内容

文献 · 验证

paper-qa

从本地论文和文档中检索证据并生成带精确引用的科研答案

查看内容

文献 · 分析 · 写作

grobid

把学术论文PDF解析成结构化TEI XML,提取元数据、全文结构和参考文献

查看内容

设计 · 分析 · 验证

CausalPy

用贝叶斯方法做准实验因果推断,含10+方法和稳健性诊断套件

查看内容

相关课程

把单项工具放进完整研究设计流程

课程一从研究问题、文献判断、理论假设和方法骨架继续推进,帮助你建立可以复用的研究设计档案。

查看课程一