def main()

in web/helpers/export.py [0:0]

• 18 lines of code
• 4 McCabe index (conditional complexity)

def main():
    """
    Load the model and train on it.
    """
    args = argument_parser().parse_args()
    OmniglotModel(args.classes, **model_kwargs(args))

    with tf.Session() as sess:
        print('var trainedParameters = [')
        tf.train.Saver().restore(sess, tf.train.latest_checkpoint(args.checkpoint))
        for conv_name in ['', '_1', '_2', '_3']:
            names = [x % conv_name for x in ['conv2d%s/kernel:0',
                                             'batch_normalization%s/gamma:0',
                                             'batch_normalization%s/beta:0']]
            for name in names:
                print_var(sess, name)
                print_var(sess, name.replace(':0', '/Adam_1:0'))
        print_var(sess, 'dense/kernel:0')
        print_var(sess, 'dense/kernel/Adam_1:0')
        print_var(sess, 'dense/bias:0')
        print_var(sess, 'dense/bias/Adam_1:0', last=True)
        print('];')