jingyimei edited a comment on issue #400: DL: modify the logic to get number of output units from model_arch table URL: https://github.com/apache/madlib/pull/400#issuecomment-497796214 > When we say `assume the last dense layer contains the right num_classes` do we have confidence this is the general way to do it? > > see comment on `layer.output_shape` in > https://stackoverflow.com/questions/49527159/how-to-get-the-output-shape-of-a-layer-in-keras @fmcquillan99 For the stackoverflow page, we can get the shape by calling layers.output_shape, but still don't know which one is the last dense layer. After discussion we believe the current solution is the best we can do for now to get the right number.
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] With regards, Apache Git Services
