python - 将 model.summary() 保存为 pdf

Question

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

我尝试按照这里提出的问题：如何将 model.summary() 保存到 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)

但是文件不可读。

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

标签： pythontensorflowpdfkeras

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)