Skip to content

Latest commit

 

History

History
14 lines (12 loc) · 521 Bytes

README.md

File metadata and controls

14 lines (12 loc) · 521 Bytes

SLVAE

The code has been tested under Pytorch 1.8. SLVAE is included in the GraphSL library.

To test the code, simply run the following command

cd <submission1>
python slvae.py -d <datasetname>

Datasets names are 'jazz_SIR', 'jazz_SI', 'cora_ml_SIR', 'cora_ml_SI', 'power_grid_SIR', 'power_grid_SI', 'karate_SIR', 'karate_SI', 'netscience_SIR', 'netscience_SI' Run the following code to see available parameters that can be passed in:

python slvae.py -h