Deep-Forest

GitHub
962 167 非常简单 1 次阅读 2周前NOASSERTION开发框架
AI 解读 由 AI 自动生成,仅供参考

Deep-Forest 是一个高效、可扩展且经过优化的 Python 框架,专为实现“深度森林”算法而设计。它旨在解决传统树模型在处理表格数据时精度受限或调参复杂的问题,为随机森林和梯度提升决策树(GBDT)等经典算法提供了一种强有力的替代方案。

该工具的核心优势在于其卓越的预测性能,往往能超越现有的集成学习方法,同时大幅降低了参数调整的门槛,让用户无需耗费大量精力即可获得优异结果。在技术层面,Deep-Forest 结合了深度学习的层级结构与树模型的鲁棒性,不仅训练速度快、效率高,还能轻松应对大规模数据集的挑战。

Deep-Forest 特别适合机器学习开发者、数据科学家以及科研人员使用,尤其是那些专注于表格数据分析、希望在不依赖深度神经网络复杂架构的前提下提升模型表现的用户。通过简洁的 API 接口,用户可以快速上手进行分类或回归任务,只需几行代码即可完成从数据加载到模型评估的全流程。如果你正在寻找一种既强大又易用的树基学习工具,Deep-Forest 值得尝试。

使用场景

某金融风控团队正在构建信用卡欺诈检测模型,需要在海量结构化交易数据中快速识别异常行为并上线生产。

没有 Deep-Forest 时

  • 调参耗时巨大:传统 GBDT 或随机森林模型对超参数极其敏感,数据科学家需花费数天进行网格搜索以寻找最优组合。
  • 精度遭遇瓶颈:面对复杂的非线性欺诈特征,现有树模型集成方法的准确率难以突破,导致漏报率居高不下。
  • 训练效率低下:随着交易数据量激增,模型训练时间显著拉长,难以满足业务对每日增量更新的时效要求。
  • 扩展性受限:在处理千万级样本的大规模数据集时,内存占用过高,常导致任务中断或需要昂贵硬件支持。

使用 Deep-Forest 后

  • 开箱即用省心:Deep-Forest 具备强大的自适应能力,大幅减少人工调参工作量,默认配置即可在多数场景下获得优异表现。
  • 预测精度跃升:凭借独特的级联森林结构,Deep-Forest 在表格数据上的分类准确率显著超越传统集成算法,有效降低欺诈漏报。
  • 训练速度飞快:优化的底层架构使得模型训练效率大幅提升,能在短时间内完成全量数据建模,支持高频迭代。
  • 轻松应对大规模数据:Deep-Forest 具有良好的可扩展性,能够流畅处理千万级行数据,无需过度依赖硬件升级。

Deep-Forest 通过“少调参、高精度、快训练”的特性,为结构化数据的建模任务提供了比传统树模型更高效的终极解决方案。

运行环境要求

操作系统
  • 未说明
GPU

未说明

内存

未说明

依赖
notes该工具是基于树的集成学习方法(Deep Forest),主要针对表格数据优化,不依赖深度学习框架(如 PyTorch 或 TensorFlow),因此通常无需 GPU。安装方式为 pip install deep-forest。若需处理图像等非结构化数据以利用多粒度扫描功能,需参考原始的 gcForest 实现。
python3.6+
scikit-learn
numpy
scipy
Deep-Forest hero image

快速开始

深森林 (DF) 21

|github|_ |readthedocs|_ |codecov|_ |python|_ |pypi|_ |style|_

.. |github| image:: https://github.com/LAMDA-NJU/Deep-Forest/workflows/DeepForest-CI/badge.svg .. _github: https://github.com/LAMDA-NJU/Deep-Forest/actions

.. |readthedocs| image:: https://readthedocs.org/projects/deep-forest/badge/?version=latest .. _readthedocs: https://deep-forest.readthedocs.io

.. |codecov| image:: https://codecov.io/gh/LAMDA-NJU/Deep-Forest/branch/master/graph/badge.svg?token=5BVXOT8RPO .. _codecov: https://codecov.io/gh/LAMDA-NJU/Deep-Forest

.. |python| image:: https://img.shields.io/pypi/pyversions/deep-forest .. _python: https://pypi.org/project/deep-forest/

.. |pypi| image:: https://img.shields.io/pypi/v/deep-forest?color=blue .. _pypi: https://pypi.org/project/deep-forest/

.. |style| image:: https://img.shields.io/badge/code%20style-black-000000.svg .. _style: https://github.com/psf/black

DF21Deep Forest <https://arxiv.org/pdf/1702.08835.pdf>__ 2021年2月1日版本的实现。它具有以下优势:

  • 强大:比现有的基于树的集成方法具有更高的准确性。
  • 易用:无需过多调参。
  • 高效:训练速度快,效率高。
  • 可扩展:能够处理大规模数据。

DF21 为随机森林或 GBDT 等基于树的机器学习算法提供了一种有效且强大的选择。

如需快速入门,请参阅 如何开始使用 <https://deep-forest.readthedocs.io/en/latest/how_to_get_started.html>。有关参数调优的详细指南,请参阅 参数调优 <https://deep-forest.readthedocs.io/en/latest/parameters_tunning.html>

DF21 针对基于树的集成方法擅长处理的任务(即表格数据)进行了优化。如果您希望利用多粒度扫描部分更好地处理图像等结构化数据,请参阅 原始实现 <https://github.com/kingfengji/gcForest>__ 以获取更多详情。

安装

DF21 可以通过 Python 的包管理工具 PyPI <https://pypi.org/project/deep-forest/>__ 使用 pip 安装。您可以使用 pip 从 Python 包索引及其他索引中安装软件包。有关 pip 的文档,请参阅 此处 <https://pypi.org/project/pip/>__。使用以下命令即可下载 DF21:

.. code-block:: bash

pip install deep-forest

快速入门

分类


.. code-block:: python

from sklearn.datasets import load_digits
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score

from deepforest import CascadeForestClassifier

X, y = load_digits(return_X_y=True)
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=1)
model = CascadeForestClassifier(random_state=1)
model.fit(X_train, y_train)
y_pred = model.predict(X_test)
acc = accuracy_score(y_test, y_pred) * 100
print("\n测试准确率:{:.3f}%".format(acc))
>>> 测试准确率:98.667%

回归


.. code-block:: python

from sklearn.datasets import load_boston
from sklearn.model_selection import train_test_split
from sklearn.metrics import mean_squared_error

from deepforest import CascadeForestRegressor

X, y = load_boston(return_X_y=True)
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=1)
model = CascadeForestRegressor(random_state=1)
model.fit(X_train, y_train)
y_pred = model.predict(X_test)
mse = mean_squared_error(y_test, y_pred)
print("\n测试均方误差:{:.3f}".format(mse))
>>> 测试均方误差:8.068

资源

  • 文档 <https://deep-forest.readthedocs.io/>__
  • Deep Forest:[会议] <https://www.ijcai.org/proceedings/2017/0497.pdf>__ | [期刊] <https://academic.oup.com/nsr/article-pdf/6/1/74/30336169/nwy108.pdf>__
  • AISTATS 2019 主题演讲:[幻灯片] <https://aistats.org/aistats2019/0-AISTATS2019-slides-zhi-hua_zhou.pdf>__

参考文献

.. code-block:: latex

@article{zhou2019deep,
    title={Deep forest},
    author={Zhi-Hua Zhou and Ji Feng},
    journal={National Science Review},
    volume={6},
    number={1},
    pages={74--86},
    year={2019}}

@inproceedings{zhou2017deep,
    title = {{Deep Forest:} Towards an alternative to deep neural networks},
    author = {Zhi-Hua Zhou and Ji Feng},
    booktitle = {IJCAI},
    pages = {3553--3559},
    year = {2017}}

感谢所有贡献者

|contributors|

.. |contributors| image:: https://contributors-img.web.app/image?repo=LAMDA-NJU/Deep-Forest .. _contributors: https://github.com/LAMDA-NJU/Deep-Forest/graphs/contributors

版本历史

v0.1.72022/10/01
v0.1.62022/09/17
v0.1.52021/04/16
v0.1.42021/03/11
v0.1.32021/02/22
v0.1.22021/02/11
v0.1.12021/02/07

常见问题

相似工具推荐

openclaw

OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你

349.3k|★★★☆☆|昨天
Agent开发框架图像

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|昨天
开发框架图像Agent

everything-claude-code

everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上

142.7k|★★☆☆☆|今天
开发框架Agent语言模型

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

107.9k|★★☆☆☆|今天
开发框架图像Agent

markitdown

MarkItDown 是一款由微软 AutoGen 团队打造的轻量级 Python 工具,专为将各类文件高效转换为 Markdown 格式而设计。它支持 PDF、Word、Excel、PPT、图片(含 OCR)、音频(含语音转录)、HTML 乃至 YouTube 链接等多种格式的解析,能够精准提取文档中的标题、列表、表格和链接等关键结构信息。 在人工智能应用日益普及的今天,大语言模型(LLM)虽擅长处理文本,却难以直接读取复杂的二进制办公文档。MarkItDown 恰好解决了这一痛点,它将非结构化或半结构化的文件转化为模型“原生理解”且 Token 效率极高的 Markdown 格式,成为连接本地文件与 AI 分析 pipeline 的理想桥梁。此外,它还提供了 MCP(模型上下文协议)服务器,可无缝集成到 Claude Desktop 等 LLM 应用中。 这款工具特别适合开发者、数据科学家及 AI 研究人员使用,尤其是那些需要构建文档检索增强生成(RAG)系统、进行批量文本分析或希望让 AI 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器

93.4k|★★☆☆☆|今天
插件开发框架

LLMs-from-scratch

LLMs-from-scratch 是一个基于 PyTorch 的开源教育项目,旨在引导用户从零开始一步步构建一个类似 ChatGPT 的大型语言模型(LLM)。它不仅是同名技术著作的官方代码库,更提供了一套完整的实践方案,涵盖模型开发、预训练及微调的全过程。 该项目主要解决了大模型领域“黑盒化”的学习痛点。许多开发者虽能调用现成模型,却难以深入理解其内部架构与训练机制。通过亲手编写每一行核心代码,用户能够透彻掌握 Transformer 架构、注意力机制等关键原理,从而真正理解大模型是如何“思考”的。此外,项目还包含了加载大型预训练权重进行微调的代码,帮助用户将理论知识延伸至实际应用。 LLMs-from-scratch 特别适合希望深入底层原理的 AI 开发者、研究人员以及计算机专业的学生。对于不满足于仅使用 API,而是渴望探究模型构建细节的技术人员而言,这是极佳的学习资源。其独特的技术亮点在于“循序渐进”的教学设计:将复杂的系统工程拆解为清晰的步骤,配合详细的图表与示例,让构建一个虽小但功能完备的大模型变得触手可及。无论你是想夯实理论基础,还是为未来研发更大规模的模型做准备

90.1k|★★★☆☆|今天
语言模型图像Agent