tensorflow2.x(2)
-
[Pytorch & Tensorflow2.x 기초] CNN 모델로 Pytorch와 Tensorflow2.x의 기본적인 모델 구현 확인하기
TensorFlow 2.0¶ In [1]: import tensorflow as tf from tensorflow.keras import layers from tensorflow.keras import datasets Hyperparameter Tunning¶ In [2]: num_epochs = 1 batch_size = 64 learning_rate = 0.001 dropout_rate = 0.7 input_shape = (28, 28, 1) num_classes = 10 Preprocess¶ In [3]: # data load (train_x, train_y), (test_x, test_y) = datasets.mnist.load_data() In [5]: # 차원 늘리기 train_x = trai..
2020.09.01 -
[tensorflow2.x 기초 - 6] MNIST data로 Evaluation & Prediction을 구현기본
TensorFlow: Evaluating & Prediction¶ In [1]: import tensorflow as tf from tensorflow.keras import layers from tensorflow.keras import datasets Build Model¶ In [2]: input_shape = (28, 28, 1) num_classes = 10 learning_rate = 0.001 In [3]: inputs = layers.Input(input_shape) net = layers.Conv2D(32, (3, 3), padding='SAME')(inputs) net = layers.Activation('relu')(net) net = layers.Conv2D(32, (3, 3), p..
2020.09.01