快捷導(dǎo)航

一文教你如何在Python中忽略煩人的警告

更新時間：2024年12月23日 09:45:48 作者：engchina

當(dāng)你用 Python 寫代碼時,有時候會看到一些“警告”信息,這些信息不會讓代碼出錯,但會讓輸出看起來很亂,所以本文為大家整理了一些忽略警告的方法,希望對大家有所幫助

當(dāng)你用 Python 寫代碼時，有時候會看到一些“警告”信息。這些信息不會讓代碼出錯，但會讓輸出看起來很亂。如果你覺得這些警告很煩，可以用下面的代碼讓它們“閉嘴”。今天我們就來學(xué)學(xué)這段代碼是怎么做到的！

代碼講解

1. 導(dǎo)入需要用到的工具

import warnings
import logging
from transformers import logging as hf_logging

warnings：這是 Python 內(nèi)置的工具，用來管理“警告”。

logging：這是 Python 的日志工具，日志就是程序運行時輸出的提示信息。

hf_logging：這是 transformers 庫的日志工具，我們也可以用它調(diào)整日志的顯示。

2. 定義一個函數(shù)：ignore_warnings

def ignore_warnings():

函數(shù)就像一個“工具箱”，當(dāng)你需要它的功能時，只要調(diào)用它就可以。

3. 忽略特定的警告

warnings.filterwarnings("ignore", message="Some weights of the model checkpoint")
warnings.filterwarnings("ignore", message="Could not find image processor class")
warnings.filterwarnings("ignore", message="The `max_size` parameter is deprecated")

warnings.filterwarnings("ignore", ...)：這是說“我要忽略某些警告”。

message：這里寫的是警告的內(nèi)容，只要警告里包含這段文字，就會被“靜音”。

就像我們告訴程序：“看到這些‘內(nèi)容’的警告，別煩我！”

4. 調(diào)整日志顯示級別

logging.basicConfig(level=logging.ERROR)

logging.basicConfig：設(shè)置日志的“聲音大小”。

level=logging.ERROR：告訴程序“只顯示出錯的信息，別展示其他沒用的提示”。

5. 設(shè)置 Transformers 庫的日志等級

hf_logging.set_verbosity_error()

hf_logging.set_verbosity_error()：這是專門用來調(diào)節(jié) transformers 庫的日志，告訴它“只顯示出錯信息，其他的不要說話”。

什么時候用這段代碼？

用深度學(xué)習(xí)工具：比如 transformers，它會輸出很多沒用的警告，影響閱讀。

調(diào)試代碼時：有時候只想看重要的信息，忽略那些不必要的提示。

總結(jié)

這段代碼就是一把“靜音遙控器”，幫我們把那些煩人的警告和不重要的信息屏蔽掉，讓代碼的輸出更加干凈整潔！

完整代碼

import warnings
import logging
from transformers import logging as hf_logging

def ignore_warnings():
    # 忽略特定的警告
    warnings.filterwarnings("ignore", message="Some weights of the model checkpoint")
    warnings.filterwarnings("ignore", message="Could not find image processor class")
    warnings.filterwarnings("ignore", message="The `max_size` parameter is deprecated")

    # 調(diào)整日志等級，只顯示錯誤信息
    logging.basicConfig(level=logging.ERROR)
    hf_logging.set_verbosity_error()

快把這段代碼加到你的項目里試試吧！它會讓你的輸出世界清靜很多哦。

到此這篇關(guān)于一文教你如何在Python中忽略煩人的警告的文章就介紹到這了,更多相關(guān)Python忽略警告內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: