Skip to content

aleksgorica/MLGproject

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

32 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MLGproject

article: https://medium.com/@nm8144/a7a40caeb959

Running

  • If running from local, set env variable DATASET_PATH to path of your local dataset folder.
  • If running from colab, just run and use Google GPU :)
  • Running GAT: very important, to make sure concat=False in GATConv layer. Otherwise, it will not work. See here for more info.

Config

There are two configs:

  • Test config: specify training configuration (epochs, batch size, etc.)
  • Model config: specify model configuration (layers, activation functions, etc.)

It's best to re-run from Model definition cell to avoid errors and unexpected behaviour.

Todo

  • DONE :)

Dataset

Is available at Kaggle

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •