Muyi 0.0.0__py3-none-any.whl → 0.0.2__py3-none-any.whl

Muyi-0.0.2.dist-info/METADATA

@@ -0,0 +1,46 @@
+Metadata-Version: 2.1
+Name: Muyi
+Version: 0.0.2
+Summary: Some useful utils.
+Home-page: https://github.com/Muyiiiii/muyi
+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
+
+# muyi
+
+Some useful utils for GNNs and Deep Learning.
+
+
+
+```bash
+pip install muyi
+```
+
+
+
+
+
+## utils
+
+1. `color_print(content)`
+2. `save_pic_iterly(pic_name, postfix, info)`
+3. `read_csv_iterly(path, **kwargs)`
+
+
+
+## graph
+
+1. `pyg_data_to_dgl_graph(pyg_data_obj)`
+
+
+
+## gpu
+
+1. `get_gpu_memory_usage()`
+2. `display_gpu_memory_usage()`

Muyi-0.0.2.dist-info/RECORD

@@ -0,0 +1,9 @@
+muyi/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+muyi/gpu.py,sha256=JLr-hQ84gv7afgFX6XHaquvN-HB19NqvwHxoq-rugdM,676
+muyi/graph.py,sha256=4okLvgp0TboSpuKtlzXTTbLin2K-RMVxhqJ7CZt6Tps,1145
+muyi/utils.py,sha256=Mulmh8T7r5YHum8CgceOURyJgvzHzCZV0FPZurmhgys,1385
+Muyi-0.0.2.dist-info/LICENSE,sha256=GzHS9pi6vw_oGwKIGxAPch8iuVrHazOilEsYL7AlPK0,1085
+Muyi-0.0.2.dist-info/METADATA,sha256=6QuPjmjJBNfALSMd5gAXN_kNc9i4XHn0neuCrLzB1C0,763
+Muyi-0.0.2.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+Muyi-0.0.2.dist-info/top_level.txt,sha256=3X0xmx0SvmCNEj1ISqmD2evKh_ejCJQaLA4GEbGfKc4,5
+Muyi-0.0.2.dist-info/RECORD,,

{Muyi-0.0.0.dist-info → Muyi-0.0.2.dist-info}/WHEEL

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

muyi/gpu.py

@@ -0,0 +1,20 @@
+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.")

muyi/utils.py

@@ -1,26 +1,7 @@
 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.")
+import pandas as pd
+from tqdm import tqdm
 
 def color_print(content):
     print(f'\033[1;46m{content}\033[0m\n')
@@ -37,3 +18,26 @@ def save_pic_iterly(pic_name, postfix, info):
     plt.savefig(pic_name_full, dpi=300, bbox_inches='tight')
 
     color_print(f'!!!!! {info} is saved in file {pic_name_full}')
+
+def read_csv_iterly(path, **kwargs):
+    INPUT_FILENAME = path
+    LINES_TO_READ_FOR_ESTIMATION = 20
+    CHUNK_SIZE_PER_ITERATION = 10**5
+
+
+    temp = pd.read_csv(INPUT_FILENAME,
+                    nrows=LINES_TO_READ_FOR_ESTIMATION, **kwargs)
+    N = len(temp.to_csv(index=False))
+    df = [temp[:0]]
+    t = int(os.path.getsize(INPUT_FILENAME)/N*LINES_TO_READ_FOR_ESTIMATION/CHUNK_SIZE_PER_ITERATION) + 1
+
+
+    with tqdm(total = t, file = sys.stdout) as pbar:
+        for i,chunk in enumerate(pd.read_csv(INPUT_FILENAME, chunksize=CHUNK_SIZE_PER_ITERATION, low_memory=False, **kwargs)):
+            df.append(chunk)
+            pbar.set_description('Importing: %d' % (1 + i))
+            pbar.update(1)
+
+    data = temp[:0].append(df)
+    del df            
+    return data

Muyi-0.0.0.dist-info/METADATA

@@ -1,16 +0,0 @@
-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

@@ -1,8 +0,0 @@
-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,,