def model_filename(self, model_name, epoch):
    return '{model_name}-epoch-{epoch}.pk'.format(model_name=model_name, epoch=epoch)
def model_filename(self, model_name, epoch):
    return '{model_name}-epoch-{epoch}.pk'.format(model_name=model_name, epoch=epoch)