GetIT-Sunday/MOSAIC - 模块化优化与搜索的自主智能体框架
一个受AutoKaggle启发的多智能体框架,用于自主数据科学竞赛。
概述
MOSAIC(Modular Optimization and Search for Agentic-,模块化优化与搜索的自主智能体框架)是一个先进的多智能体系统,旨在自动化端到端的数据科学竞赛流程。受AutoKaggle的启发,MOSAIC通过将数据科学工作流分解为专门的智能体模块,实现高效的协作问题解决。
核心特性
- 模块化架构:基于可组合的智能体模块(数据探索、特征工程、模型选择、超参数优化等)
- 智能搜索与优化:集成贝叶斯优化、遗传算法等高级搜索策略
- 多智能体协作:支持智能体之间的消息传递、任务委派和结果聚合
- 竞赛平台支持:针对Kaggle等数据科学竞赛平台优化
- 可扩展性:允许用户自定义智能体角色、工作流和外部工具集成
快速开始
安装
git clone https://github.com/GetIT-Sunday/MOSAIC-Modular-Optimization-and-Search-for-Agentic-.git
cd MOSAIC-Modular-Optimization-and-Search-for-Agentic-
pip install -r requirements.txt
基本用法
- 配置竞赛环境并设置API密钥(如Kaggle API)
- 定义智能体工作流
- 运行框架
from mosaic import MOSAICAgent
agent = MOSAICAgent()
agent.run(competition="titanic")
架构概览
MOSAIC的核心架构由以下主要组件构成:
- 智能体管理器(Agent Manager):负责智能体的生命周期管理、任务分派和通信协调
- 模块注册中心(Module Registry):存储所有可用的智能体模块及其配置
- 搜索优化引擎(Search Optimizer):实现超参数搜索、特征选择和模型集成策略
- 结果聚合器(Result Aggregator):从多个智能体收集结果并生成最终解决方案
配置
通过YAML或JSON文件配置智能体、模块和竞赛参数:
agent:
type: ensemble
modules:
- data_explorer
- feature_engineer
- model_trainer
search:
method: bayesian
max_iterations: 100
贡献
欢迎提交Issue和Pull Request。请遵循现有的代码风格和测试规范。
许可
本项目采用MIT许可协议。详情请参阅LICENSE文件。