python 将model.summary()另存为pdf会导致文件不可读

qv7cva1a 于 2023-02-28 发布在 Python

关注(0)|答案(3)|浏览(177)

我正在尝试将我的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)

但文件无法读取。
我问到这里，巴尔加夫给了我如下的解决方案：

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

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

当用Google Docs打开时看起来很好，但当以PDF格式打开/下载时只会给出错误，就好像文件被损坏了一样。
由于我有几个CNN模型包括在我的文档中，我需要他们自动在样式和格式，我希望他们在。
得到一个LaTEX友好的表也可以，但我需要有人给予我一个MWE，因为我不知道如何做到这一点。

3条答案

您是否需要将摘要保存为PDF？如果需要，您需要使用fpdf或PyPDF2等 Package 器库来生成所需的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)

最后我发现其他人已经生成了一种相对方便的方法来将模型摘要转换为LaTeX表，这将很好地满足我的目的。
溶液here

我用这个

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

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