# 智图 (AutoGL) [English Introduction](../..) 用于图数据的自动机器学习框架和工具包。 *由清华大学媒体与网络实验室进行开发与维护* 若有任何意见或建议，欢迎通过issues 或邮件autogl@tsinghua.edu.cn与我们联系。 ## 最新消息 - 2022.12.30 新版本！智图v0.4.0-pre 版本发布! - 提出了**NAS-Bench-Graph** ([paper](https://openreview.net/pdf?id=bBff294gqLp),[code](https://github.com/THUMNLab/NAS-Bench-Graph), [tutorial](http://mn.cs.tsinghua.edu.cn/autogl/documentation/docfile/tutorial/t_nas_bench_graph.html))，并发表于NeurIPS'22上！其是首个针对图数据的神经架构搜索Benchmark。将AutoGL与NAS-Bench-Graph一起使用可以极大加快图神经架构搜索算法的性能评估过程 - 支持图自动机器学习的**鲁棒性**算法!包括图结构特征工程、鲁棒图神经网络模型和鲁棒图神经架构搜索，详情参见[图鲁棒性教程](http://mn.cs.tsinghua.edu.cn/autogl/documentation/docfile/tutorial_cn/t_robustness.html) - 支持图自动机器学习的**自监督**算法！详情参见[图自监督教程](http://mn.cs.tsinghua.edu.cn/autogl/documentation/docfile/tutorial_cn/t_ssl_trainer.html) - 2021.12.31 智图发布v0.3.0-pre版本： - 支持**[Deep Graph Library (DGL)](https://www.dgl.ai/)**作为后端，包括同构图的节点分类、链接预测以及图分类等任务。智图现在也可兼容PyG 2.0版本。 - 支持**异构图**节点分类任务!详情请参考[异构图教程](http://mn.cs.tsinghua.edu.cn/autogl/documentation/docfile/tutorial/t_hetero_node_clf.html)。 - `model`模块目前支持**解耦**为两个子模块，即编码器`encoder`和解码器`decoder`，一个`encoder`可以被用来处理不同任务。 - 扩展了支持的[神经架构搜索算法](http://mn.cs.tsinghua.edu.cn/autogl/documentation/docfile/tutorial/t_nas.html)，例如[AutoAttend](https://proceedings.mlr.press/v139/guan21a.html)，[GASSO](https://proceedings.neurips.cc/paper/2021/hash/8c9f32e03aeb2e3000825c8c875c4edd-Abstract.html)， [硬件感知算法](http://mn.cs.tsinghua.edu.cn/autogl/documentation/docfile/documentation/nas.html#autogl.module.nas.estimator.OneShotEstimator_HardwareAware)等。 - 2021.07.11 智图v0.2.0-pre版本发布，并支持了[神经架构搜索(NAS)](http://mn.cs.tsinghua.edu.cn/autogl/documentation/docfile/tutorial/t_nas.html)，可以对给定的数据集和架构定制化神经网络架构；[采样](http://mn.cs.tsinghua.edu.cn/autogl/documentation/docfile/tutorial/t_trainer.html#node-classification-with-sampling)以处理大规模图数据集；和链接预测任务。 - 2021.04.16 我们在IJCAI 2021发表了关于图自动机器学习的首个综述文章! 详情见[这里](http://arxiv.org/abs/2103.00742)。 - 2021.04.10 我们的论文**[AutoGL: A Library for Automated Graph Learning](https://arxiv.org/abs/2104.04987)**已经发表于 _ICLR 2021 Workshop on Geometrical and Topological Representation Learning_ ! ## 介绍智图的设计目标是可以简单、快速地对图数据集和任务进行自动机器学习，可供研究者和开发者使用。更多详细信息，可以参阅我们的文档。下图是智图的整体框架。

智图通过 `datasets` 类以支持图数据集，其基于 PyTorch Geometric 和 Deep Graph Library 的数据集，并添加了一些函数以支持自动机器学习框架。智图通过 `AutoGL solvers` 以处理不同的图机器学习任务，利用五个主要模块自动解决给定的任务，即自动特征工程 `auto feature engineer`，神经架构搜索 `neural architecture search`，自动模型 `auto model`，超参数优化 `hyperparameter optimization`，和自动模型集成 `auto ensemble`。目前，智图支持以下算法：

特征工程	图模型	神经架构搜索	超参数优化	模型集成
生成器 Graphlets EigenGNN 更多 ... 选择器 SeFilterConstant gbdt 全图特征 Netlsd NxAverageClustering 更多 ...	同构图编码器 GCNEncoder GATEncoder SAGEEncoder GINEncoder 解码器 LogSoftmaxDecoder DotProductDecoder SumPoolMLPDecoder JKSumPoolDecoder	搜索算法 Random RL Evolution GASSO 更多 ... 搜索空间 SinglePath GraphNas AutoAttend 更多 ... 模型评估 Oneshot Scratch	Grid Random Anneal Bayes CAMES MOCAMES Quasi random TPE AutoNE	Voting Stacking

此工具包还可作为一个框架供用户实现和测试自己的自动机器学习或图机器学习模型。 ## 安装 ### 依赖在安装智图之前，请首先安装以下依赖项。 1. Python >= 3.6.0 2. PyTorch (>=1.6.0) 详细信息请参考。 3. 图机器学习工具包智图需要 PyTorch Geometric（PyG）或 Deep Graph Library（DGL）作为后端。若两者均安装，可在运行时选择任一后端，参考[这里](http://mn.cs.tsinghua.edu.cn/autogl/documentation/docfile/tutorial/t_backend.html)。 3.1 PyTorch Geometric (>=1.7.0) 详细信息请参考。 3.2 Deep Graph Library (>=0.7.0) 详细信息请参考。 ### 安装 #### 通过pip进行安装运行以下命令以通过`pip`安装智图。 ``` pip install autogl ``` #### 从源代码安装运行以下命令以从源安装智图。 ``` git clone https://github.com/THUMNLab/AutoGL.git cd AutoGL python setup.py install ``` #### 开发者安装如果您想以开发者方式安装智图，请运行以下命令以创建软链接，然后即可修改本地程序后而无需重复安装。 ``` pip install -e . ``` ## 文档您可参考文档页面以参阅我们的详细文档。文档也可以进行本地编译。首先，请安装 sphinx 和 sphinx-rtd-theme: ``` pip install -U Sphinx pip install sphinx-rtd-theme ``` 然后，通过以下方式创建 html 文档： ``` cd docs make clean && make html ``` 文档将在如下路径自动生成：`docs/_build/html` ## 引用如果您使用了智图代码，请按如下方式引用我们的[论文](https://openreview.net/forum?id=0yHwpLeInDn): ``` @inproceedings{guan2021autogl, title={Auto{GL}: A Library for Automated Graph Learning}, author={Chaoyu Guan and Ziwei Zhang and Haoyang Li and Heng Chang and Zeyang Zhang and Yijian Qin and Jiyan Jiang and Xin Wang and Wenwu Zhu}, booktitle={ICLR 2021 Workshop on Geometrical and Topological Representation Learning}, year={2021}, url={https://openreview.net/forum?id=0yHwpLeInDn} } ``` 或许您也会发现我们的[综述](http://arxiv.org/abs/2103.00742)有帮助: ``` @article{zhang2021automated, title={Automated Machine Learning on Graphs: A Survey}, author={Zhang, Ziwei and Wang, Xin and Zhu, Wenwu}, booktitle = {Proceedings of the Thirtieth International Joint Conference on Artificial Intelligence, {IJCAI-21}}, year={2021}, note={Survey track} } ``` ## 版权相关从v0.2版本开始，智图的所有代码采用[Apache license](LICENSE)。 ## 用户交流微信群 ![weixin](https://user-images.githubusercontent.com/17705534/163801846-2aa1bdf8-919e-4636-b7d9-e5346f352c15.jpg)