能否介绍下多标签多分类问题吗?
来源:7-3 实战准备
MNSF
2022-02-03
写回答
1回答
-
flare_zhao
2022-02-09
多分类主要的不同就是在输出层,根据类别数量确定输出层神经元个数,同时将输出层的激活函数修改为适合多分类问题的softmax函数即可。比如十分类代码:
#建立mlp模型
from keras.models import Sequential
from keras.layers import Dense, Activation
mlp = Sequential()
mlp.add(Dense(units = 392, input_dim = 784, activation='relu'))
mlp.add(Dense(units = 196, activation='relu'))
mlp.add(Dense(units=10, activation='softmax'))
mlp.summary()
注意:多分类问题,输入数据的label,要转化为one-hot数据类型
00
相似问题