TorchCP

给深度学习预测加上可证明的可靠性边界,一行代码搞定不确定性量化。

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 开源模型 🍴 复刻 52 🐛 待处理 12 👀 关注 8
GitHub ↗
GitHub 星标 ★ 482
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,LGPL-3.0许可,完全免费,可自行部署使用。

项目预览

TorchCP 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类开源模型
开发团队ml-stat-Sustech
定价模式🏷️完全免费
价格说明开源项目,LGPL-3.0许可,完全免费,可自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议LGPL-3.0
主要语言Python
技术栈/模型classification,conformal-prediction,graph,large-language-models,regression,uncertainty-quantification
GitHub 星标★ 482
上线时间2023-12-06 00:00:00
代码更新2026-08-05
仓库体积10.0 MB
本站数据更新2026-11-09
维护状态低维护
收录时间2026-08-09
浏览次数6

项目介绍

TorchCP 是一个基于 PyTorch 的保形预测(Conformal Prediction)研究工具箱,专为深度学习模型设计。它解决了深度学习模型在输出预测时缺乏严格不确定性量化保证的问题,为研究者提供了一套统一、易用的接口来实现各类保形预测方法。核心能力包括:支持多种主流保形预测算法(如 Split CP、CV+、Jackknife+ 等)、提供与 PyTorch 模型无缝集成的 API、内置常见回归和分类任务的评估指标与可视化工具。工具箱强调模块化设计,用户可灵活组合预测器、校准器和评分函数,快速复现论文结果或验证新方法。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 482

使用教程

难度:入门 约 10 分钟 部署方式:本地安装 3 步

环境要求

  • Python
  • PyTorch

安装与启动步骤

  1. 1直接装包

    已发布到包索引,一条命令装完。

    pip install torchcp
  2. 2跑一个置信预测示例

    先用自带示例确认接口形态。

  3. 3换自己的模型输出

    把分类器的 logits 换成自己的即可。

如何确认成功

用自己的模型输出跑一次,能得到带覆盖率保证的预测集合,且覆盖率随显著性水平变化

注意事项

  • 它是保形预测的工具箱,做的是给已有模型输出加统计保证,不是训练模型。
  • 核心概念是覆盖率与预测集合大小,用之前先理解这两者的取舍。
  • 接口设计贴着 PyTorch,接自己的模型只需换数据来源。

核心亮点

  • 模块化设计,预测器/校准器/评分函数可自由组合,实验配置灵活
  • 内置多种经典保形算法(Split CP、CV+、Jackknife+),开箱即用
  • 与 PyTorch 原生集成,无需改造现有模型即可接入

不足之处

  • 早期项目,文档和示例相对有限
  • 目前主要支持回归和分类任务,对结构化输出覆盖不足

适用场景

  • 学术研究中对比不同保形预测方法的实证效果
  • 工业场景下为深度学习模型输出附加可靠性区间,辅助风险决策
  • 教学场景中演示不确定性量化原理与算法实现

替代项目

MAPIE、crepes、nonconformist

上一篇:Finetuning-Suite

下一篇:BlaGPT

同类项目推荐

mrmr 开源

高维数据下自动选特征,去冗余提精度

mRMR (minimum-Redundancy-Maximum-Relevance) for automatic feature selection at scale···

★ 631 2026-08-09
ai-toolkit 开源

写份配置就能微调扩散模型,LoRA 训练开箱即用

The ultimate training toolkit for finetuning diffusion models

★ 12586 2026-10-12
OpenThoughts-Agent 开源

一套配方搞定 Agent 训练数据与流程

Data recipes and robust infrastructure for training AI agents

★ 310 2026-10-17
ETO 开源

让 LLM Agent 在试错中自我进化,越用越聪明

Trial and Error: Exploration-Based Trajectory Optimization of LLM Agents (ACL 2024 M···

★ 170 2026-08-09
hf_ehr 开源

用 EHR 数据微调 HuggingFace 模型,快速构建医疗预测应用

Training HuggingFace models on EHR data

★ 55 2026-08-09
Merge-Stable-Diffusion-models-wi··· 开源

不训练,无损合并多个Stable Diffusion模型,统一风格与能力。

Adaptation of the merging method described in the paper - Git Re-Basin: Merging Mode···

★ 147 2026-08-09
alignment-handbook 开源

一键复现主流 LLM 对齐实验,从 SFT 到 DPO 全流程开箱即用。

Robust recipes to align language models with human and AI preferences

★ 5711 2026-08-09
BitVAE 开源

用比特级分词,让图像生成更精细高效

official training and inference code of bitwise tokenizer

★ 73 2026-08-09
dpm-solver 开源

10步内完成扩散模型采样,图像生成提速数倍

Official code for "DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model S···

★ 1860 2026-08-09