파이썬/Code Solution

[Code Solution] Tensorflow, Keras 모델 동일한 결과 얻기

컴공돌이​ 2022. 6. 20. 12:34

Tensorflow, Keras 모델 동일한 결과 얻기

Deep Learning 모델을 구축할 때 Tensorflow, Keras 라이브러리를 많이 사용하고 있습니다.

하지만, Deep Learning 모델의 특성상 학습할 때마다 결과가 달라질 수 있습니다.

따라서, 동일한 결과를 얻기 위하여 random seed를 설정해 주어야 합니다.

import tensorflow as tf
import numpy as np
import random

random_seed = 42
random.seed(random_seed)
np.random.seed(random_seed)
tf.set_random_seed(random_seed)

python version: 3.7.13

 

※ 만약 위의 코드와 같이 random seed를 설정해주었음에도 불구하고 여전히 동일한 결과가 나오지 않는다면 모델이 멀티 쓰레드를 사용하고 있거나, 코드 상에서 사용하고 있는 외부 라이브러리에서 random_seed를 사용하고 있을 수 있습니다.

반응형