Variable Misnomer outlayer of Generator

mp2893 / medgan

Generative adversarial network for generating electronic health records.

BSD 3-Clause "New" or "Revised" License

270 stars 90 forks source link

Variable Misnomer outlayer of Generator #16

Closed lnyemba closed 5 years ago

lnyemba commented 5 years ago

Hi guys, I think there's an issue in the output layer of the generator on line 99. I believe it is a variable misnomer from the execution graph (I hope I am wrong)

I think it should be

        W = tf.get_variable('W_'+str(i), shape=[tempDim, self.generatorDims[-1]])

instead of

        W = tf.get_variable('W'+str(i), shape=[tempDim, self.generatorDims[-1]])