Replaced the batch sizes with 'None' instead '?'

I have replaced the default text to be shown if batch sizes are not present as 'None' instead of '?' because it is more readable and understandable during the plotting of the model. Furthermore, it avoids the confusion of any error in the model and signifies that the fact that the programmer voluntarily haven't given any batch sizes
This commit is contained in:
Vividha 2020-07-26 23:39:47 +05:30 committed by GitHub
parent 3a3d359192
commit 3f4e4cb19d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -206,7 +206,7 @@ def model_to_dot(model,
if show_shapes:
def format_shape(shape):
return str(shape).replace(str(None), '?')
return str(shape).replace(str(None), 'None')
try:
outputlabels = format_shape(layer.output_shape)