Muyi 0.0.0__py3-none-any.whl

Muyi-0.0.0.dist-info/LICENSE

@@ -0,0 +1,16 @@
+Copyright (c) 2018 The Python Packaging Authority
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

Muyi-0.0.0.dist-info/METADATA

@@ -0,0 +1,16 @@
+Metadata-Version: 2.1
+Name: Muyi
+Version: 0.0.0
+Summary: Some useful utils.
+Author: muyiiiii
+Author-email: 
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3
+Description-Content-Type: text/markdown
+License-File: LICENSE
+
+# muyiiiii_graph
+
+Some useful utils for GNNs.

Muyi-0.0.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+muyi/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+muyi/graph.py,sha256=4okLvgp0TboSpuKtlzXTTbLin2K-RMVxhqJ7CZt6Tps,1145
+muyi/utils.py,sha256=VuuZ0S4CieON-bNHmKqrinTEvmIdfUQF0gXhiapkUnM,1220
+Muyi-0.0.0.dist-info/LICENSE,sha256=GzHS9pi6vw_oGwKIGxAPch8iuVrHazOilEsYL7AlPK0,1085
+Muyi-0.0.0.dist-info/METADATA,sha256=FGffGdElQ4CcBWlVtWz-eu4cnOboef_YGz2ZJhodRhE,400
+Muyi-0.0.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+Muyi-0.0.0.dist-info/top_level.txt,sha256=3X0xmx0SvmCNEj1ISqmD2evKh_ejCJQaLA4GEbGfKc4,5
+Muyi-0.0.0.dist-info/RECORD,,

Muyi-0.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: bdist_wheel (0.43.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

Muyi-0.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+muyi

muyi/graph.py

@@ -0,0 +1,42 @@
+import torch
+import dgl
+
+def pyg_data_to_dgl_graph(pyg_data_obj):
+    print(pyg_data_obj)
+
+    # 获取边索引
+    edge_index = pyg_data_obj.edge_index
+    
+    # DGL需要的边索引格式是两列的数组，而不是两行的索引
+    src, dst = edge_index
+    edge_list = torch.stack((src, dst), dim=1)
+    
+    # 创建DGL图
+    g = dgl.graph((edge_list[:, 0], edge_list[:, 1]), num_nodes=pyg_data_obj.x.shape[0])
+    
+    # 添加节点特征
+    if 'x' in pyg_data_obj:
+        g.ndata['feature'] = pyg_data_obj.x
+    
+    # 添加边特征
+    if 'edge_attr' in pyg_data_obj:
+        g.edata['feat'] = pyg_data_obj.edge_attr
+    
+    # 添加节点标签
+    if 'y' in pyg_data_obj:
+        g.ndata['label'] = pyg_data_obj.y
+
+    # 添加节点标签
+    if 'train_mask' in pyg_data_obj:
+        g.ndata['train_mask'] = pyg_data_obj.train_mask
+
+    # 添加节点标签
+    if 'test_mask' in pyg_data_obj:
+        g.ndata['test_mask'] = pyg_data_obj.test_mask
+
+    # 添加节点标签
+    if 'val_mask' in pyg_data_obj:
+        g.ndata['val_mask'] = pyg_data_obj.val_mask
+
+    
+    return g

muyi/utils.py

@@ -0,0 +1,39 @@
+import os
+import matplotlib.pyplot as plt
+
+import GPUtil
+
+def get_gpu_memory_usage():
+    # 获取GPU使用情况
+    gpus = GPUtil.getGPUs()
+    gpu_memory_info = []
+    for gpu in gpus:
+        gpu_memory_info.append((gpu.id, gpu.name, f"{gpu.memoryUsed} MB", f"{gpu.memoryTotal} MB", f"{gpu.memoryUtil * 100:.1f}%"))
+
+    return gpu_memory_info
+
+def display_gpu_memory_usage():
+    gpu_memory_info = get_gpu_memory_usage()
+
+    if gpu_memory_info:
+        print("GPU Memory Usage:")
+        for info in gpu_memory_info:
+            print(f"GPU ID: {info[0]}, Name: {info[1]}, Memory Used: {info[2]}, Total Memory: {info[3]}, Memory Utilization: {info[4]}")
+    else:
+        print("No GPU found.")
+
+def color_print(content):
+    print(f'\033[1;46m{content}\033[0m\n')
+
+def save_pic_iterly(pic_name, postfix, info):
+    pic_idx=1
+    pic_name_full=f'{pic_name}_{pic_idx}.{postfix}'
+
+    while os.path.exists(pic_name_full):
+        print(f'File {pic_name_full} already exists.')
+        pic_idx += 1
+        pic_name_full=f'{pic_name}_{pic_idx}.png'
+
+    plt.savefig(pic_name_full, dpi=300, bbox_inches='tight')
+
+    color_print(f'!!!!! {info} is saved in file {pic_name_full}')