快捷導(dǎo)航

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

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

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

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

代碼講解

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

import warnings
import logging
from transformers import logging as hf_logging

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

logging：這是 Python 的日志工具，日志就是程序運(yùn)行時(shí)輸出的提示信息。

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

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

def ignore_warnings():

函數(shù)就像一個(gè)“工具箱”，當(dāng)你需要它的功能時(shí)，只要調(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)容，只要警告里包含這段文字，就會(huì)被“靜音”。

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

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

logging.basicConfig(level=logging.ERROR)

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

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

5. 設(shè)置 Transformers 庫(kù)的日志等級(jí)

hf_logging.set_verbosity_error()

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

什么時(shí)候用這段代碼？

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

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

總結(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)整日志等級(jí)，只顯示錯(cuò)誤信息
    logging.basicConfig(level=logging.ERROR)
    hf_logging.set_verbosity_error()

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

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

您可能感興趣的文章: