[딥러닝] 딥러닝 국소해의 문제 해결해보기

 

인공신경망 학습이 제대로 되지 않는 경우 국소해 문제로 발생합니다.

국소해 문제가 발생할 경우엔 재학습을 수행해보거나 은닉층의 노드수를 변경해봅니다.

import tensorflow as tf
from _7seg_data import X, YT

model=tf.keras.Sequential([
    tf.keras.Input(shape=(7,)),
    tf.keras.layers.Dense(16, activation='relu'),
    tf.keras.layers.Dense(4, activation='sigmoid')
])

model.compile(optimizer='adam',loss='mse')

model.fit(X,YT,epochs=10000)

Y=model.predict(X)
print(Y)