CNN 예제
-
Tensorflow Convolution Neural Network 예시 코드(CNN, MNIST DataSet)Data Science/Tensorflow 2022. 10. 26. 22:04
1. 기본 세팅 1.1 라이브러리 import tensorflow as tf import numpy as np import matplotlib.pyplot as plt import random import os from sklearn.model_selection import train_test_split from tensorflow.keras import Model, layers 1.2 랜덤시드 고정 매번 동일한 결과를 가져오기 위한 랜덤시드 고정 random.seed(1) np.random.seed(1) tf.random.set_seed(1) 1.3 GPU 세팅 os.environ["CUDA_VISIBLE_DEVICES"]="0" gpus = tf.config.experimental.list_physi..