TensorFlow로 첫 신경망 만들기

이 단원에서는 TensorFlow를 사용해 간단한 피드포워드 신경망을 만들어 보겠습니다. TensorFlow는 머신러닝 모델을 구축하고 학습시키는 데 널리 사용되는 프레임워크입니다.

1단계: TensorFlow 설정

먼저 TensorFlow가 설치되어 있는지 확인합니다. 설치되어 있지 않다면 다음 명령을 사용해 설치할 수 있습니다:

pip install tensorflow

또한 이 단원에 필요한 라이브러리를 가져옵니다.

import tensorflow as tf
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense
import numpy as np

2단계: 데이터셋 준비

이 예시에서는 입력이 특성 하나이고 출력이 그 값의 두 배인 간단한 데이터셋을 사용합니다:

# Generate sample data
X = np.array([[1], [2], [3], [4], [5]], dtype=float)
y = np.array([[2], [4], [6], [8], [10]], dtype=float)

3단계: 모델 만들기

입력층 하나, 은닉층 하나, 출력층 하나로 구성된 간단한 모델을 만들어 보겠습니다. 은닉층에서는 ReLU 활성화 함수를 사용합니다.

# Create the model
model = Sequential([
    Dense(10, activation='relu', input_shape=(1,)),  # Hidden layer with 10 neurons
    Dense(1)  # Output layer with 1 neuron
])

4단계: 모델 컴파일

다음과 같이 모델을 컴파일합니다:

• 손실 함수: 회귀 작업에는 평균 제곱 오차를 사용합니다.
• 최적화기: 확률적 경사 하강법(SGD)을 사용합니다.

# Compile the model
model.compile(optimizer='sgd', loss='mean_squared_error')

5단계: 모델 훈련

훈련 데이터를 사용하여 모델을 500 에포크 동안 훈련합니다. 훈련하는 동안 모델은 손실을 최소화하는 방법을 학습합니다.

# Train the model
model.fit(X, y, epochs=500, verbose=1)

6단계: 예측 수행

훈련이 끝나면 모델을 사용하여 새로운 데이터에 대한 예측을 수행할 수 있습니다:

# Predict on new data
print(model.predict([6]))

7단계: 모델 평가

모델이 패턴을 올바르게 학습했는지 확인하기 위해 훈련 데이터에서 모델의 성능을 평가합니다:

# Evaluate the model
loss = model.evaluate(X, y)
print(f"Loss: {loss}")

실제 활용 사례

TensorFlow로 구축한 순전파 신경망은 다음과 같은 분야에 활용할 수 있습니다:

• 예측 분석: 판매량과 추세 예측
• 분류: 스팸 탐지, 사기 탐지
• 회귀: 가격 예측, 수요 예측

요약 및 다음 단계

이번 레슨에서는 다음을 수행했습니다:

• TensorFlow를 사용하여 간단한 신경망을 구축하고 훈련했습니다.
• 데이터 집합 준비, 모델 생성, 컴파일, 훈련이라는 주요 단계를 학습했습니다.
• 예측을 수행하고 모델의 성능을 평가했습니다.

다음에는 이미지 처리 작업을 위한 합성곱 신경망과 같은 더 고급 구조로 신경망을 확장하는 방법을 살펴봅니다.