I'm reading your paper and have a question about section 3.2. Could you explain what are the vertex matrix and context matrix? Are they simply the source and the destination of edges?

Python has a test package which can shadow the import of the test.py module, leading to an ImportError when importing NN or kNN.

https://docs.python.org/2.7/library/test.html?highlight=test#module-test
https://docs.python.org/3.5/library/test.html

I suggest to rename it to test_embedding.py or something like that.

Describe the bug
In current master branch, we calculate the GTC by calling srl_zoo/plotting/representation_plot.py. In particular, in the function plotCorrelation, we call numpy to compute the correlation.

corr = np.corrcoef(x=x + eps, y=states_rewards["states"] + eps, rowvar=0)

However in the [numpy.corrcoef doc](https://docs.scipy.org/doc/numpy/reference/generated/nu

Just a small observation to point out that the requirement.txt when executed on virtual environment installs a CPU backed tensorflow instead of a GPU one.
Better, change it to tensorflow-gpu==1.5 or point out that this give you a CPU tensorflow.

Thanks !!

I viewed the whole code and found that the code only use toy dummy data to train model. So I don't really understand how you use those data to train GCN model. Can you supply the code or instructions about how to use real-world data to train model?