TensorFlow는 머신러닝을 위한 엔드 투 엔드 오픈소스 플랫폼입니다.
TensorFlow를 사용하면 초보자와 전문가 모두 머신러닝 모델을 쉽게 만들 수 있습니다. 시작하려면 아래의 섹션을 참조하세요.
import tensorflow as tf
mnist = tf.keras.datasets.mnist
(x_train, y_train),(x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0
model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(input_shape=(28, 28)),
tf.keras.layers.Dense(128, activation='relu'),
tf.keras.layers.Dropout(0.2),
tf.keras.layers.Dense(10, activation='softmax')
])
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
model.fit(x_train, y_train, epochs=5)
model.evaluate(x_test, y_test)
전문가용
Subclassing API는 고급 연구에 사용할 수 있는 실행 시 정의되는(define-by-run) 인터페이스를 제공합니다. 모델 클래스를 만든 다음 전달 패스를 명령형으로 만드세요. 맞춤 레이어, 활성화, 학습 루프를 손쉽게 작성할 수 있습니다. 아래의 'Hello World' 예제를 실행한 다음 튜토리얼을 방문하여 자세한 내용을 알아보세요.
class MyModel(tf.keras.Model):
def __init__(self):
super(MyModel, self).__init__()
self.conv1 = Conv2D(32, 3, activation='relu')
self.flatten = Flatten()
self.d1 = Dense(128, activation='relu')
self.d2 = Dense(10, activation='softmax')
def call(self, x):
x = self.conv1(x)
x = self.flatten(x)
x = self.d1(x)
return self.d2(x)
model = MyModel()
with tf.GradientTape() as tape:
logits = model(images)
loss_value = loss(logits, labels)
grads = tape.gradient(loss_value, model.trainable_variables)
optimizer.apply_gradients(zip(grads, model.trainable_variables))
일반적인 문제 해결책
프로젝트에 도움이 되는 단계별 튜토리얼을 탐색하세요.
Keras를 사용한 ML 기본사항
첫 번째 신경망
빠르게 진행되는 전체 TensorFlow 프로그램 개요에서 운동화 및 셔츠와 같은 의류 이미지를 분류하도록 신경망을 학습시키세요.
