기본 형식
import tensorflow as tf from tensorflow import keras
model = keras.Sequential()
model.add(넣고싶은 레이어)
model.add(넣고싶은 레이어)
model.add(넣고싶은 레이어)
model.fit(x, y, epochs=10, batch_size=32)
- 모델의 입출력이 여러 개인 경우 적절하지 않음.
### 2) Functional API
```python
import tensorflow as tf
from tensorflow import keras
inputs = keras.Input(shape=(x,y,z)) (x*y*z)의 차원으로 정의됨
x = keras.layers.__넣고싶은 레이어__(채널 수, 커널의 크기, activation=활성화 함수 종류)(input)
x = keras.layers.__넣고싶은 레이어__(관련 파라미터)(x)
outputs = keras.layers.__넣고싶은 레이어__(관련 파라미터)(x)
model = keras.Model(inputs=inputs, outputs=outputs)
model.fit(x,y, epochs=10, batch_size=32)
import tensorflow as tf
from tensorflow import keras
class CustomModel(keras.Model):
def __init__(self):
super().__init__()
self.conv1 = keras.layers.Conv2D(32, 3, activation='relu')
self.conv2 = keras.layers.Conv2D(64, 3, activation='relu')
self.flatten = keras.layers.Flatten()
self.fc1 = keras.layers.Dense(128, activation='relu')
self.fc2 = keras.layers.Dense(10, activation='softmax')
def call(self, x):
x = self.conv1(x)
x = self.conv2(x)
x = self.flatten(x)
x = self.fc1(x)
x = self.fc2(x)
return x
model = CustomModel()
model.fit(x,y, epochs=10, batch_size=32)
tensorflow v2 api에 대한 개념을 학습한 후 알고리즘을 풀이했다. 기본적인 문제는 익숙하게 풀 수 있는데 조금이라도 응용되면 시간도 오래 걸리고 이해도 금방하지 못해서 사고력을 키울 수 있게 계속 단계적으로 생각하는 연습을 해야할 것 같다.