README.md

GraphPrepKit 🕸🛟

GraphPrepKit 🕸🛟 is a powerful and versatile graph preprocessing library designed to simplify and streamline the data preparation phase for graph-based applications. It provides a comprehensive set of tools and utilities to efficiently preprocess graphs, allowing users to transform, clean, and manipulate graph data with ease.

Wiki 📚

👉 Documentation Here 📑

Installation ⚙️

pip install -q git+'https://github.com/p4zaa/GraphPrepKit.git'

from GraphPrepKit import construct, homo, hetero

Graph Dataset Construction 🛠️

Construct connection dataframe

Note target column must contain list of connected elements. Example

source	target
't0001'	['u0001', 'u0005', 'u0010']
't0002'	['u0001', 'u0008', 'u0012']

>>> dfConn = construct.get_connection_table(df, target='target')
>>> dfConn

source	target
't0001'	'u0001'
't0001'	'u0005'
't0001'	'u0010'
't0002'	'u0001'
't0002'	'u0008'
't0002'	'u0012'

Map unique ID to its associated index

>>> map_to_idx(dfConn, source='source', target='target', type='hetero', inplace=True)
>>> dfConn

source	target
0	0
0	1
0	3
1	0
1	2
1	4