Clear Filters
Clear Filters

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

6 views (last 30 days)
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)

Accepted Answer

yanqi liu
yanqi liu on 1 Mar 2022
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 Comments
yanqi liu
yanqi liu on 1 Mar 2022
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'}
>>

Sign in to comment.

More Answers (0)

Categories

Find more on Deep Learning Toolbox in Help Center and File Exchange

Products


Release

R2021b

Community Treasure Hunt

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

Start Hunting!