大三寒假学习进度(8)

时间:2021-01-27 13:05:18   收藏:0   阅读:0

1. 使用八股搭建神经网络

1.1 、 用Tensorflow API: tf.keras 搭建神经网络八股

# impor相关模块
import tensorflow as tf
from sklearn import datasets
import numpy as np

# 交代训练集的输入特征x_train和训练集的标签y_train
x_train = datasets.load_iris().data
y_train = datasets.load_iris().target

# 测试集的输入特征x_test和测试集标签y_test可以直接给定,也可以在fit中按比例从训练集中划分

#实现了数据集的乱序
np.random.seed(116)
np.random.shuffle(x_train)
np.random.seed(116)
np.random.shuffle(y_train)
tf.random.set_seed(116)

#在Sequential()搭建网络结构
model = tf.keras.models.Sequential([
    tf.keras.layers.Dense(3, activation=‘softmax‘, kernel_regularizer=tf.keras.regularizers.l2())
])

# 在compile中配置训练方法
model.compile(optimizer=tf.keras.optimizers.SGD(lr=0.1),
              loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=False),
              metrics=[‘sparse_categorical_accuracy‘])

#在fit()中执行训练过程
model.fit(x_train, y_train, batch_size=32, epochs=500, validation_split=0.2, validation_freq=20)

# 打印网络结构和参数统计
model.summary()
评论(0
© 2014 mamicode.com 版权所有 京ICP备13008772号-2  联系我们:gaon5@hotmail.com
迷上了代码!