GNN Model with 93% Accuracy for Facebook Page-Page Network Node Classification with TSNE Visualization - #169

Open

liammulhern wants to merge 27 commits into

shakes76:mainfrom

liammulhern:topic_GNN_SSC

Commits on Oct 3, 2024

Adds initial README with project scope
liammulhern
committed
ba7e30b

Commits on Oct 26, 2024

Adds initial README for GNN with project scope
liammulhern
committed
ba785bb
Init pattern recognition python files
liammulhern
committed
76d53ab
Adds file descriptions for pattern recognition files
liammulhern
committed
e4f82e2
Adds test and util file descriptions, updates gitignore for model files
liammulhern
committed
eb91268
Adds FLPP dataset loader and test cases to validate features, classes, and nodes
liammulhern
committed
3b34f27
Adds FLPP dataset loader with test training split
liammulhern
committed
efd0159
Adds graph visualisation and related tests
liammulhern
committed
1b77d5c

Commits on Oct 27, 2024

Adds edge list to adjacency matrix using torch geo
liammulhern
committed
342677f
Adds GNN and sparse layer modules
liammulhern
committed
559c989
Adds train and test methods for generic model inputs
liammulhern
committed
45ef881
Adds multiple epoch train method that saves and loads model
liammulhern
committed
a44ee7f
Updates save and load to device functionality for model values
liammulhern
committed
bcb5afa
Adds initial model inference for categorisation of graph from GNN
liammulhern
committed
bbfd83c
Adds rangpur slurm runner script
liammulhern
committed
04c5bd2
Adds entry point with CLI args for running model functions; Adds full GNN training method
liammulhern
committed
e76f5b2
Adds TSNE from raw FLPP dataset and figure output
liammulhern
committed
d2c160f
Updates dataloader for validate split
liammulhern
committed
1b40b44

Commits on Oct 28, 2024

Adds adjacency matrix to training model and begins readme documentation
liammulhern
committed
b09d81e
Updates script to ease training setup
liammulhern
committed
2adf099
Removes pytorch geo dependency
liammulhern
committed
45b57b5
Adds dataset, hyperparameters, and model figures to readme
liammulhern
committed
9e3ba4c
Adds overview and implementation documentation to readme
liammulhern
committed
2b15848
Adds trained TSNE plot and updates training loader to use tensor masks
liammulhern
committed
dc42071
Adds accuracy and loss plots for trained model
liammulhern
committed
4e38681
Adds inference interface to main.py
liammulhern
committed
c5e3d98
Adds conclusion and discussion to readme file
liammulhern
committed
a495aec