使用Tensorflow keras创建GRU

最新推荐文章于 2023-05-27 21:25:54 发布

风华明远

最新推荐文章于 2023-05-27 21:25:54 发布

阅读量2.8k

点赞数 1

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

本文链接：https://blog.csdn.net/weixin_42272768/article/details/106582596

版权

本文介绍了如何在TensorFlow的Keras API中创建GRU（门控循环单元）神经网络。通过两种方式——直接使用GRU层以及通过RNN层配合GRUCell来实现GRU网络。

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

创建GRU也有两种方法：GRU和RNN+GRUCell

import tensorflow as tf
import numpy as np
from tensorflow import keras
import os
import matplotlib.pyplot as plt

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '1'
#读取本地mnist数据
def my_load_data(path='mnist.npz'):
    origin_folder = 'https://storage.googleapis.com/tensorflow/tf-keras-datasets/'
    path = tf.keras.utils.get_file(
        path,
        origin=origin_folder + 'mnist.npz',
        cache_dir='DataSet/',
        cache_subdir=""
    )
    with np.load