GCN on Facebok Large Page-Page Network dataset (s4722208)#137

Open

jiwhan5 wants to merge 34 commits into

shakes76:topic-recognitionfrom

jiwhan5:topic-recognition

Commits on Oct 15, 2024

Initialisation
Jace
committed
GNN modules initialised
Jace
committed
GNN updated (4 convolution layers)
Jace
committed
data_load updated
Jace
committed
load_data modified to return masks

Jace
authored and
Jace
committed
hyper parameter, data loading, model and optimizer initialised for training

Jace
authored and
Jace
committed
train loop with model updated

Jace
authored and
Jace
committed
validation added to the training loop

Jace
authored and
Jace
committed
edges error fixed (edges shape [342004, 2] to [2,342004])

Jace
authored and
Jace
committed
accuracy calculation error fixed

Jace
authored and
Jace
committed
testing added

Jace
authored and
Jace
committed
TSNE plot function added

Jace
authored and
Jace
committed

Commits on Oct 16, 2024

TSNE plotting error fixed
Jace
committed
prediction based on saved model updated
Jace
committed
accuracy and loss function plotting added, TSNE plotting funciton moved from train.py
Jace
committed
calculating loss added to validation process for plotting
Jace
committed
Document of the project uploaded

Jace
authored and
Jace
committed
saved GCN model

Jace
authored and
Jace
committed
Comments added to each files

Jace
authored and
Jace
committed

Commits on Oct 17, 2024

plot_pre_train_TSNE added to plot the original data

Jace
authored and
Jace
committed
Figures uploaded

Jace
authored and
Jace
committed

Commits on Oct 25, 2024

Commits on Oct 27, 2024

Commits on Nov 19, 2024

"GCN.pth file removed"
jiwhan5
authored