causalmt 0.1.0

首个公开版本。面向多值干预决策的因果表示学习与组合干预推荐工具包。

安装

pip install causalmt

PyPI: https://pypi.org/project/causalmt/0.1.0/

核心组件

• GACRNet —— 多值干预因果效应估计，扩展 DragonNet 到 K 类干预
  • GPS 路由注意力、多尺度 MMD 表示正则、目标正则化双稳健 ATE
  • MLP / Transformer backbone，uplift / separate outcome head
  • sklearn 风格 API：fit / estimate_ate / estimate_cate / predict_potential_outcomes / save / load
• IDCIO —— 交互效应分解的组合干预推荐
  • 三阶段：原子效应估计 → 交互效应 MLP → QUBO 求解
  • from_estimator（配套 GACRNet）/ from_effects（解耦外部效应）两种构造
  • QUBO 穷举 + 贪心求解器，MC Dropout 推荐置信度

其他

• 数据加载器（IHDP / multi_attribution）、评估指标（PEHE / ATE error / policy risk）
• 3 个端到端示例 notebook
• 86 个测试，覆盖率 89%