(案例):利用bp神经网络预测蚊子种类,第三列目标值为0.9的属于不咬人的A类,目标值为0.1的属于咬人的B类,特征集第一列是翼长,第二列是触角长
如下有20个样本数据:
data=[[1.78,1.14,0.9],
[1.96,1.18,0.9],
[1.86,1.20,0.9],
[1.72,1.24,0.1],
[2.00,1.26,0.9],
[1.82,1.54,0.1],
[1.96,1.30,0.9],
[1.74,1.36,0.1],
[1.64,1.38,0.1],
[1.82,1.38,0.1],
[1.90,1.38,0.1],
[1.70,1.40,0.1],
[1.70,1.48,0.1],
[1.72,1.33,0.1],
[1.70,1.40,0.1],
[1.72,1.40,0.1],
[1.70,1.40,0.1],
[1.82,1.48,0.1],
[2.00,1.08,0.9],
[2.08,1.33,0.1]]
导包
from sklearn.preprocessing import MinMaxScaler
import numpy as np
import pandas as pd
import tensorflow as tf
对标签值进行二值化
df01 = pd.get_dummies(pd.DataFrame(data),columns=[2],sparse=False)
data = df01.values
划分训练集和测试集
data_train=data[:int