본문 바로가기
Developing../Python

모델성능비교 Tip : warning제거 하는 법

by bents 2022. 7. 18.

## print 출력 format 통합

: time, metric( ex. mIoU )

## warnings ignore 사용하기

import warnings

warnings.filterwarnings('ignore')

## print hiding class 생성 + 숨길 코드에 반영

- 단점 : 실행이 오래 걸려서 오류난 것처럼 보일 수 있음

 

- session으로 관리하기

  :  __enter__, __exit__()를 override하기

import sys, os

class HiddenPrints:
    def __enter__(self):
        self._original_stdout = sys.stdout
        sys.stdout = open(os.devnull, 'w')

    def __exit__(self, exc_type, exc_val, exc_tb):
        sys.stdout.close()
        sys.stdout = self._original_stdout
        
 
with HiddenPrints():
    run_python_func()