sklearn-diagnose

给 sklearn 模型做 AI 体检,一眼揪出过拟合和数据泄漏

sklearn-diagnose 是一个面向 Scikit-learn 模型的 AI 诊断工具,旨在帮助数据科学家快速识别模型训练中的常见问题,如过拟合、数据泄漏、类别不平衡等。它通过分析模型训练过程中的关键指标(如学习曲线、特征重要性、混淆矩阵等),结合 LLM 生成自然语言的洞察和建议,降低调试门槛。核心能力包括自动检测过拟合和欠拟合、识别数据泄漏风险、评估类别不平衡影响,并提供可操作的改进建议。该工具以 Python 库的形式集成到现有工作流中,支持与 Jupyter Notebook 无缝协作,适合模型开发阶段的快速健康检查。

开源 free 基础设施
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 62
维护状态 低维护
是否开源 是
定价模式 free

项目数据

分类基础设施
开发团队leockl
所属国家
官网地址
定价模式free
价格说明开源项目,MIT许可证,完全免费,可自行部署使用。
访问状态
是否开源是
开源协议MIT
主要语言Python
技术栈/模型ai,data-science,deep-learning,langchain,llm,machine-learning,mlops,model-diagnostics,overfitting,python,scikit-learn,sklearn
GitHub 星标★ 62
30天Star增速
HF 下载量
上线时间2026-01-07 00:00:00
最近更新2026-06-01 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数6

使用教程

核心亮点

  • 利用 LLM 将复杂诊断结果转化为通俗易懂的自然语言建议
  • 覆盖过拟合、数据泄漏、类别不平衡等高频问题,实用性强
  • 与 Scikit-learn 无缝集成,使用简单,适合快速迭代

不足之处

  • 项目处于早期,文档和社区支持有限
  • 依赖 LLM API,可能产生额外成本且需网络连接

适用场景

  • 模型训练后快速检查是否存在过拟合或数据泄漏
  • 在 Jupyter Notebook 中交互式分析模型健康度
  • 作为机器学习管道的自动化质量门禁

替代项目

Yellowbrick、scikit-learn 内置的 learning_curve 和 validation_curve、Sweetviz

项目介绍

sklearn-diagnose 是模型训练领域的开源项目,由 leockl 开发,是 2026 年新上线的项目。

它收录于模型训练分类,该分类目前共有 518 个项目,GitHub 星标数为 62。

项目已超过三个月没有代码更新,维护节奏明显放缓,最近一次代码更新于 2026-06-01。MIT许可证,完全免费,可自行部署使用。

它主要面向的使用场景是:模型训练后快速检查是否存在过拟合或数据泄漏。同类可对比的替代方案包括 Yellowbrick、scikit-learn 内置的 learning_curve 和 validation_curve、Sweetviz。

上一篇:plexe

下一篇:distributed-ml-patterns

同类项目推荐

Long-RL 开源

让强化学习轻松驾驭超长序列,训练更稳更快

Long-RL: Scaling RL to Long Sequences (NeurIPS 2025)

★ 729 2026-08-09
TPA 开源

把注意力复杂度从平方降到线性,长序列不再卡顿

[NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (https://arxiv.or···

★ 463 2026-08-09
minimind 开源

两小时造出你的小模型,练手入门不求人

Train a 64M-parameter LLM from scratch in just 2h!

★ 62178 2026-08-09
pruna 开源

一键优化 AI 模型,推理提速又省资源。

Pruna is a model optimization framework built for developers, enabling you to delive···

★ 1286 2026-08-13