How to assign a class name after model prediction using Neural Network

6 次查看(过去 30 天)
Hello everyone ,i hope you are doing well. I have trained a Feedforward neural network using pattern recognition app. I have five classes the model gives the output in vec2ind form .
I want to output to be in string form for example.
if the model predict class=4 i want to print 'Rectangle Class'
if model class=3 i want to print 'Circle Class' similary for all 5 classes
I have attched the model and the test file which predict class=4
Can anybody help me in that please
load('FNNmodel.mat')
net=trainedmodel.Network
y=net(dataset21');
classes=vec2ind(y)

采纳的回答

yanqi liu
yanqi liu 2022-3-1
if the model predict class=4 i want to print 'Rectangle Class'
if model class=3 i want to print 'Circle Class' similary for all 5 classes
load('FNNmodel.mat')
net=trainedmodel.Network
y=net(dataset21');
base_db = {'class1','class2','Circle Class','Rectangle Class','class5'};
classes=base_db{vec2ind(y)}
classes =
'Rectangle Class'
  3 个评论
yanqi liu
yanqi liu 2022-3-1
load('FNNmodel.mat')
load('dataset31')
net=trainedmodel.Network
y=net(dataset21');
base_db = {'class1','class2','Circle Class','Rectangle Class','class5'};
classes=base_db(vec2ind(y))
classes =
1×3 cell 数组
列 1 至 2
{'Rectangle Class'} {'class5'}
列 3
{'Circle Class'}
>>

请先登录,再进行评论。

更多回答(0 个)

类别

Help CenterFile Exchange 中查找有关 Deep Learning Toolbox 的更多信息

产品


版本

R2021b

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by