PyG scikit-learn

Stochastic-Graph-assisted-Genre-Classification

Code to reproduce and extend the results of the report Stochastic-Graph-assisted-Genre-Classification

Run the code

The easiest way to get started is by looking at the notebook Stochastic-Graph-assisted-Genre-Classification.ipynb

Requirements & Installation

Run the following commands. Install the following requirements, preferably via anaconda:

torch pyg sklearn

Also, go change directory

$ cd Stochastic-Graph-assisted-Genre-Classification

Finally, install the package

$ pip install -e .

Data

You can find the data used for the experiments here

The file is expected to be placed in datasets/book_review, so please create the directory if it does not exist.

Contact

Please contact andreasjosef.binder@tum.de in case you have any questions.