keiserlab / keras-neural-graph-fingerprint

Keras implementation of Neural Graph Fingerprints as proposed by Duvenaud et al., 2015
MIT License
46 stars 24 forks source link

Improve neighbour lookup performance #9

Closed tivaro closed 7 years ago

tivaro commented 7 years ago

Make sure NGF.layers.batch_lookup works faster and also works in TensorFlow by using reshape as in the TimeDistributed layer