search

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

Add trainable weights #2

Closed tivaro closed 8 years ago

tivaro commented 8 years ago

Currently the inner dense layers in NeuralGraphHidden and NeuralGraphOutput do initialise their weights, but the parameters are not stored in NeuralGraphHidden.trainable_weights. As a result, the layers are not learning.