Skip to content

landrarwolf/ML-SGNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ML-SGNN

Source code for ... "Semantic Graph Neural Network with Multi-measure Learning for Semi-supervised Classification"

Environment Settings

  • python == 3.7
  • Pytorch == 1.1.0
  • Numpy == 1.16.2
  • SciPy == 1.3.1
  • Networkx == 2.4
  • scikit-learn == 0.21.3

Usage

python main.py -d dataset -l labelrate
  • dataset: including [citeseer, uai, acm, BlogCatalog, flickr, coraml], required.
  • labelrate: including [20, 40, 60], required.

e.g.

python main.py -d citeseer -l 20

Data

Link

Usage

Please first unzip the data folders and then use. The files in folders are as follows:

citeseer/
├─citeseer.edge: edge file.  
├─citeseer.feature: feature file.  
├─citeseer.label: label file.  
├─testL/C.txt: test file. L/C, i.e., Label pre Class, L/C = 20, 40, 60.   
├─trainL/C.txt: train file. L/C, i.e., Label pre Class, L/C = 20, 40, 60.  
└─knn
   └─ck.txt: feature graph file. k = 2~9

Parameter Settings

Recorded in ./ML-SGNN/config/[L/C][dataset].ini
e.g. ./ML-SGNN/config/20citeseer.ini

  • Model_setup: parameters for training AM-GCN, such as nhid1, nhid2, beta, theta...
  • Data_setting: dataset setttings, such as paths for input, node numbers, feature dimensions...

Reference

...

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages