能否介绍下多标签多分类问题吗?

来源: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数据类型

0
0

Python3入门人工智能 掌握机器学习+深度学习

人工智能基础全方位讲解,构建完整人工智能知识体系,带你入门AI

1946 学习 · 594 问题

查看课程