一个简单的matlab项目实现三层神经网络的简单应用

一、设计目的：进行1-100以内的质数分类

二、设计思路：

1、生成1-100以内的数和对应二进制

2、对质数部分进行label为1，其余为0

3、选择前60组作为training数据，后40组testing

4、选择三层神经网络，其中hidden和output部分使用sigmoid函数

三、代码实现：

1、测试数据生成函数

function f = dataset_generator bits_num = 7; prime_table = [2,3,5,7,11,13,17,19,23,29,31,37,41,43,47,53,59,61,67,71,73,79,83,89,97]; prime_numble = 25; prime_dataset = zeros(100,8); %Generate prime number dataset 1-100 for count = 1:100 bin_str = dec2bin(count,bits_num); for i = 1:bits_num prime_dataset(count,i) = str2num(bin_str(i)); end for i = 1:prime_numble if(count == prime_table(i)) prime_dataset(count,bits_num 1) = 1; end end if(prime_dataset(count,bits_num 1)~=1) prime_dataset(count,bits_num 1) = 0;