Neural Network 3 鸢尾花神经网络搭建

最新推荐文章于 2021-06-17 22:38:52 发布

Sxyinn

最新推荐文章于 2021-06-17 22:38:52 发布

阅读量171

点赞数

分类专栏：神经网络文章标签： tensorflow

本文链接：https://blog.csdn.net/weixin_43956102/article/details/105663613

版权

本文介绍了使用TensorFlow搭建神经网络来分类鸢尾花数据。主要代码包括随机数生成和变量定义，利用正态分布标准差和随机种子确保可重复性。

摘要由CSDN通过智能技术生成

1.主代码


#导入所需模块
import tensorflow as tf
from sklearn import datasets
from matplotlib import pyplot as plt
import numpy as np

#导入数据
x_data = datasets.load_iris().data
y_data = datasets.load_iris().target

#随机打乱数字，随即种子可以保证生成的随机数一样
#shuffle 对列表进行乱序
np.random.seed(116)
np.random.shuffle(x_data)
np.random.seed(116)
np.random.shuffle(y_data)
tf.random.set_seed(116)

#f分为测试
x_train = x_data[:-30]
y_train = y_data[:-30]
x_test = x_data[-30:]
y_test = y_data[-30:]

#数据类型转换
x_train = tf.cast(x_train , tf.float32)
x_test = tf.cast(x_test , tf.float32)

#from_tensor_slices 是让数据和特征一一对应，按照batch输入
train_db = tf.data.Dataset.from_tensor_slices((x_train,y_train)).batch(32)
test_db = tf.