AI/DeepLearning

딥러닝_다중 분류 , 선형 회귀, 로지스틱 회귀에 따른 activation, loss 설정

oaho 2023. 4. 11. 15:25
반응형

딥러닝 layers 라이브러리 불러오기

from tensorflow.keras.layers import Dense, Dropout

 

▪분류 모델
model.add(Dense(3, activation='softmax'))
model.compile(loss='categorical_crossentropy', metrics=['accuracy'], optimizer='adam')

▪ 선형 회귀
model.add( Dense(1, activation='linear'))
model.compile(loss='mse', optimizer='adam', metrics=['mse', 'mae'])

▪ 로지스틱 회귀
model.add(Dense(1, activation='sigmoid'))
model.compile(loss='binary_crossentropy', metrics=['accuracy'], optimizer='adam')

반응형