[tensorflow 2] loss, optimizer 유동적으로 가져오기

tensorflow 2에는 tensorflow.keras.losses.get과 tensorflow.keras.optimizers.get이라는 함수가 있다

(1 버전에서 있는지는 확인 안해봄)

 

간단하게 예제로 설명하면 아래와 같다

 

from tensorflow.keras.losses import get as loss_get
from tensorflow.keras.optimizers import get as optimizer_get

loss = loss_get('binary_crossentropy')
optimizer = optimizer_get(class_name = 'adam', config={'learning_rate': 0.001, 'epsilon': 0.0001})

 

위와 같이 tensorflow 1 혹은 sequental model을 사용할 때의 키워드를 사용하여 string으로 불러올 수 있다.