Maxima's Lab

안녕하세요, 오늘은 Tensorflow 2에서 Classification 모델 학습 시 Callback을 Customize 하는 방법에 대해서 알아보겠습니다. MNIST 데이터 셋을 활용하여, 모델 학습하는 예시 입니다. import tensorflow as tf from tensorflow.keras.datasets import mnist from tensorflow.keras.models import Sequential from tensorflow.keras.layers import Flatten, Dense from tensorflow.keras.callbacks import Callback import numpy as np class SimpleCallBack(Callback): def __ini..