将 model.summary() 保存为 pdf

Question

我正在尝试将我的 model.summary() 打印为 pdf。

我尝试按照此处提出的问题进行操作：How to save model.summary() to file in Keras?

def myprint(s):
    with open('/content/drive/My Drive/xxx/yyy/zzz.pdf','w') as f:
        print(s, file=f)

model.summary(print_fn=myprint)

但是文件无法读取。

有人能给我一个简单的工作解决方案吗？提前致谢。

Answer 1

您需要将摘要保存为 PDF 吗？如果是这样，您需要使用 fpdf 或 PyPDF2 之类的包装库来生成所需的 PDF 元数据，以便正确格式化文件。

否则，您可以使用您的方法写入 .txt 文件：

def myprint(s):
    with open('/content/drive/My Drive/xxx/yyy/zzz.txt','w') as f:
        print(s, file=f)

model.summary(print_fn=myprint)

Answer 2

我用这个

with open('modelsummary.pdf', 'w') as f:

    model.summary(print_fn=lambda x: f.write(x + '\n'))