Muyi 0.0.0__tar.gz

muyi-0.0.0/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/Muyi.egg-info/PKG-INFO

@@ -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/Muyi.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+LICENSE
+README.md
+pyproject.toml
+setup.py
+Muyi.egg-info/PKG-INFO
+Muyi.egg-info/SOURCES.txt
+Muyi.egg-info/dependency_links.txt
+Muyi.egg-info/top_level.txt
+muyi/__init__.py
+muyi/graph.py
+muyi/utils.py

muyi-0.0.0/Muyi.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

muyi-0.0.0/Muyi.egg-info/top_level.txt

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

muyi-0.0.0/PKG-INFO

@@ -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/README.md

@@ -0,0 +1,3 @@
+# muyiiiii_graph
+
+Some useful utils for GNNs.

muyi-0.0.0/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-0.0.0/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}')

muyi-0.0.0/pyproject.toml

@@ -0,0 +1,3 @@
+[build-system]
+requires = ["setuptools>=42", "wheel"]
+build-backend = "setuptools.build_meta"

muyi-0.0.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

muyi-0.0.0/setup.py

@@ -0,0 +1,24 @@
+import setuptools
+with open("README.md", "r") as fh:
+    long_description = fh.read()
+setuptools.setup(
+    name="Muyi",  # 模块名称
+    version="0.0.0",  # 当前版本
+    author="muyiiiii",  # 作者
+    author_email="",  # 作者邮箱
+    description="Some useful utils.",  # 模块简介
+    long_description=long_description,  # 模块详细介绍
+    long_description_content_type="text/markdown",  # 模块详细介绍格式
+    # url="",  # 模块github地址
+    packages=setuptools.find_packages(),  # 自动找到项目中导入的模块
+    # 模块相关的元数据
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+    ],
+    # 依赖模块
+    install_requires=[
+    ],
+    python_requires='>=3',
+)