借助matlab神经网络工具箱实现蠓虫分类

问题





神经网络模型

因为是一个简单的二分类问题，输出的神经元设为1，隐含层神经元设为2，输出结果以0—1代表两类，由于输入有两个变量，因此可建立一个2—2—1的神经网络模型如下所示：



matlab代码

p2=[1.78       1.14
1.96       1.18
1.86       1.20
2.00       1.26
2.00       1.28
1.96       1.30
];
p1=[1.72       1.24
1.74       1.36
1.64        1.38
1.82        1.38
1.90        1.38
1.70        1.40
1.82        1.48
1.82        1.54
2.08        1.56
];
p=[p1;p2]';
pr=minmax(p);
goal=[ones(1,9),zeros(1,6)];
plot(p1(:,1),p1(:,2),'h',p2(:,1),p2(:,2),'o')
net=newff(pr,[2,1],{'logsig','logsig'});
net.trainParam.show = 10;
net.trainParam.lr = 0.05;
net.trainParam.goal = 1e-10;
net.trainParam.epochs = 50000;
net = train(net,p,goal);
x=[1.80 1.24;1.84 1.28;2.04 1.40]';
y0=sim(net,p)
y=sim(net,x)
hold on;
p11=[1.80 1.24;1.84 1.28;2.04 1.40];
plot(p11(:,1),p11(:,2),'r*');
hold on;plotpc(net.iw{1},net.b{1})

运行结果：红色点为待分类样本，蓝色点为af，绿色点为apf。





可见，触角和翼长为(l.24,1.80)、(l.28，1.84)被判定为apf类，（1.40，2.04）被判定为af类。