[Tensorflow] TypeError: evaluate() got an unexpected keyword argument 'return_dict' 문제

TypeError: evaluate() got an unexpected keyword argument 'return_dict'

2.1 버전에서 문제가 된다고 2.2 로 업그레이드 하라는 글을 봤는데, 작년 글이었고 이 후 더 많은 버전이 나왔습니다.

 

b_eval_dict = b_model.evaluate(img_test, label_test, return_dict=True)

파라메터에 return_dict 라는게 있는데 True 이면 결과를 dict 형태로 주는거 같은데, 저 파라메터가 없어졌나봅니다.

 

return_dict=True 가 되는 버전일 경우 

eval_dict = b_model.evaluate(img_test, label_test, return_dict=True)
for key,value in eval_dict.items():
    print(f'{key}: {value}')

"""
loss: 0.3484964966773987
accuracy: 0.8737000226974487
"""

 

 

return_dict 가 없는 버전일 경우

list 로 반환 되기 때문에

eval_dict = b_model.evaluate(img_test, label_test)
for value in eval_dict:
    print(f'{value}')

"""
0.3484964966773987
0.8737000226974487
"""

첫번째가 loss, 두번째가 accuracy 네요.