.. _dataset_cn:

AutoGL 鏁版嵁闆�
==============

鎴戜滑鍩轰簬PyTorch-Geometric (PyG)锛孌eep Graph Learning (DGL)鍙奜pen Graph Benchmark (OGB)绛夊浘瀛︿範搴撴彁渚涗簡澶氱澶氭牱鐨勫父鐢ㄦ暟鎹泦銆�
鍚屾椂锛岀敤鎴峰彲浠ヤ娇鐢ˋutoGL鎵€鎻愪緵鐨勭粺涓€闈欐€佸浘瀹瑰櫒``GeneralStaticGraph``鑷畾涔夐潤鎬佸悓鏋勫浘鍙婂紓鏋勫浘锛屼緥濡傦細

.. code-block:: python

    from autogl.data.graph import GeneralStaticGraph, GeneralStaticGraphGenerator

    ''' 鍒涘缓鍚屾瀯鍥� '''
    custom_static_homogeneous_graph = GeneralStaticGraphGenerator.create_homogeneous_static_graph(
        {'x': torch.rand(2708, 3), 'y': torch.rand(2708, 1)}, torch.randint(0, 1024, (2, 10556))
    )

    ''' 鍒涘缓寮傛瀯鍥� '''
    custom_static_heterogeneous_graph = GeneralStaticGraphGenerator.create_heterogeneous_static_graph(
        {
            'author': {'x': torch.rand(1024, 3), 'y': torch.rand(1024, 1)},
            'paper': {'feat': torch.rand(2048, 10), 'z': torch.rand(2048, 13)}
        },
        {
            ('author', 'writing', 'paper'): (torch.randint(0, 1024, (2, 5120)), torch.rand(5120, 10)),
            ('author', 'reading', 'paper'): torch.randint(0, 1024, (2, 3840)),
        }
    )


鎻愪緵鐨勫父鐢ㄦ暟鎹泦
----------------
AutoGL鐩墠鎻愪緵濡備笅澶氱甯哥敤鍩哄噯鏁版嵁闆嗭細

鍗婄洃鐫ｈ妭鐐瑰垎绫伙細

+------------------+------------+-----------+--------------------------------+
| 鏁版嵁闆�           |  PyG       |  DGL      |  榛樿train/val/test鍒掑垎        |
+==================+============+===========+================================+
| Cora             | 鉁�          | 鉁�         | 鉁�                              |
+------------------+------------+-----------+--------------------------------+
| Citeseer         | 鉁�          | 鉁�         | 鉁�                              |
+------------------+------------+-----------+--------------------------------+
| Pubmed           | 鉁�          | 鉁�         | 鉁�                              |
+------------------+------------+-----------+--------------------------------+
| Amazon Computers | 鉁�          | 鉁�         |                                |
+------------------+------------+-----------+--------------------------------+
| Amazon Photo     | 鉁�          | 鉁�         |                                |
+------------------+------------+-----------+--------------------------------+
| Coauthor CS      | 鉁�          | 鉁�         |                                |
+------------------+------------+-----------+--------------------------------+
| Coauthor Physics | 鉁�          | 鉁�         |                                |
+------------------+------------+-----------+--------------------------------+
| Reddit           | 鉁�          | 鉁�         | 鉁�                              |
+------------------+------------+-----------+--------------------------------+
| ogbn-products    | 鉁�          | 鉁�         | 鉁�                              |
+------------------+------------+-----------+--------------------------------+
| ogbn-proteins    | 鉁�          | 鉁�         | 鉁�                              |
+------------------+------------+-----------+--------------------------------+
| ogbn-arxiv       | 鉁�          | 鉁�         | 鉁�                              |
+------------------+------------+-----------+--------------------------------+
| ogbn-papers100M  | 鉁�          | 鉁�         | 鉁�                              |
+------------------+------------+-----------+--------------------------------+


鍥惧垎绫讳换鍔★細 MUTAG, IMDB-Binary, IMDB-Multi, PROTEINS, COLLAB绛�

+-------------+------------+------------+--------------+------------+--------------------+
|  鏁版嵁闆�     | PyG        | DGL        | 鑺傜偣鐗瑰緛     | 鏍囩       | 杈圭壒寰�             |
+=============+============+============+==============+============+====================+
| MUTAG       | 鉁�          | 鉁�          |  鉁�           | 鉁�          | 鉁�                  |
+-------------+------------+------------+--------------+------------+--------------------+
| IMDB-Binary | 鉁�          | 鉁�          |              | 鉁�          |                    |
+-------------+------------+------------+--------------+------------+--------------------+
| IMDB-Multi  | 鉁�          | 鉁�          |              | 鉁�          |                    |
+-------------+------------+------------+--------------+------------+--------------------+
| PROTEINS    | 鉁�          | 鉁�          |  鉁�           | 鉁�          |                    |
+-------------+------------+------------+--------------+------------+--------------------+
| COLLAB      | 鉁�          | 鉁�          |              | 鉁�          |                    |
+-------------+------------+------------+--------------+------------+--------------------+
| ogbg-molhiv | 鉁�          | 鉁�          |  鉁�           | 鉁�          | 鉁�                  |
+-------------+------------+------------+--------------+------------+--------------------+
| ogbg-molpcba| 鉁�          | 鉁�          |  鉁�           | 鉁�          | 鉁�                  |
+-------------+------------+------------+--------------+------------+--------------------+
| ogbg-ppa    | 鉁�          | 鉁�          |              | 鉁�          | 鉁�                  |
+-------------+------------+------------+--------------+------------+--------------------+
| ogbg-code2  | 鉁�          | 鉁�          |  鉁�           | 鉁�          | 鉁�                  |
+-------------+------------+------------+--------------+------------+--------------------+


閾炬帴棰勬祴浠诲姟锛氱洰鍓岮utoGL鍙互浣跨敤閽堝鑺傜偣鍒嗙被浠诲姟鐨勫绉嶅浘鏁版嵁杩涜鑷姩閾炬帴棰勬祴銆�

閫氳繃GeneralStaticGraph搴忓垪鏋勫缓鑷畾涔夋暟鎹泦
----------------------------------------------------------------
濡備笅浠ｇ爜鐗囨灞曠ず浜嗛€氳繃涓€涓敱``GeneralStaticGraph``搴忓垪鏋勫缓鑷畾涔夋暟鎹泦鐨勬柟娉曘€�

.. code-block:: python
    from autogl.data import InMemoryDataset
    ''' graphs鍙橀噺鏄竴涓敱GeneralStaticGraph瀹炰緥鎵€鏋勬垚鐨勫簭鍒� '''
    graphs = [ ... ]
    custom_dataset = InMemoryDataset(graphs)