Skip to content

🎲 PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)

License

Notifications You must be signed in to change notification settings

imsouza/pytorch-geometric-temporal-dcrnn

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Installation

First install pytorch and pytorch-geometric and then run

pip install torch-geometric-temporal

Adaptation to make

  1. Edit '/torch_geometric_temporal/dataset/custom.py' to your desired database name

  2. Create and upload a json for the dataset, needs to contain a few values that can be modified but here is a preview:

{"edges": [[0, 0], [1, 1], [1, 0], [0, 1]], 
"edge_weights": [1.0, 1.0, 1.0, 1.0],
"node_ids": {"node0_name": 0, "node1_name": 1}, 
"datababase": [[]]
}
  1. After created and uploaded the json database, edit this funcion on custom.py
def _read_web_data(self):
    url = ""  # DEFINE JSON URL HERE, TIPS: UPLOAD JSON TO GITHUB (can be on this project, path:dataset) AND COPY RAW URL+
    self._dataset = json.loads(urllib.request.urlopen(url).read())
  1. If you edited the name of the file custom.py and Class, change on 'torch_geometric_temporal\dataset_init_.py':
#...
from .custom import CustomDatasetLoader  # if you edit the custom.py file, you need to restart the kernel and update name here
#...
  1. Check I for possible changes and this is the code you need to run for tests and results.

  2. After all edits and changes DO THE NEXT SECTION "Before Executing run"

  3. Run 'examples\recurrent\dcrnn_example.py'


Before Executing run on root folder

python setup.py build

and

python setup.py install

About

🎲 PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages

  • Python 100.0%