zoukankan      html  css  js  c++  java
  • Tensorflow基础教程7:关于Keras Pipeline

      目录

      Keras Sequential/Functional API 模式建立模型

      使用 Keras Model 的 `compile` 、 `fit` 和 `evaluate` 方法训练和评估模型

      前面几章的示例均使用了 Keras 的 Subclassing API 建立模型,即对 tf.keras.Model 类进行扩展以定义自己的新模型,同时手工编写了训练和评估模型的流程。这种方式灵活度高,且与其他流行的深度学习框架(如 PyTorch、Chainer)共通,是推荐的。不过在很多时候,我们只需要建立一个结构相对简单和典型的神经网络(比如上文中的 MLP 和 CNN),并使用常规的手段进行训练。这时,Keras 也给我们提供了另一套更为简单高效的内置方法来建立、训练和评估模型。

      Keras Sequential/Functional API 模式建立模型

      最典型和常用的神经网络结构是将一堆层按特定顺序叠加起来,那么,我们是不是只需要提供一个层的列表,就能由 Keras 将它们自动首尾相连,形成模型呢?Keras 的 Sequential API 正是如此。通过向 tf.keras.models.Sequential() 提供一个层的列表,就能快速地建立一个 tf.keras.Model 模型并返回:

      model = tf.keras.models.Sequential([

      tf.keras.layers.Flatten(),

      tf.keras.layers.Dense(100, activation=tf.nn.relu),

      tf.keras.layers.Dense(10),

      tf.keras.layers.Softmax()

      ])

      不过,这种层叠结构并不能表示任意的神经网络结构。为此,Keras 提供了 Functional API,帮助我们建立更为复杂的模型,例如多输入 / 输出或存在参数共享的模型。其使用方法是将层作为可调用的对象并返回张量(这点与之前章节的使用方法一致),并将输入向量和输出向量提供给 tf.keras.Model 的 inputs 和 outputs 参数,示例如下:

      inputs = tf.keras.Input(shape=(28, 28, 1))

      x = tf.keras.layers.Flatten()(inputs)

      x = tf.keras.layers.Dense(units=100, activation=tf.nn.relu)(x)

      x = tf.keras.layers.Dense(units=10)(x)

      outputs = tf.keras.layers.Softmax()(x)

      model = tf.keras.Model(inputs=inputs, outputs=outputs)

      使用 Keras Model 的 compile 、 fit 和 evaluate 方法训练和评估模型

      当模型建立完成后,通过 tf.keras.Model 的 compile 方法配置训练过程:

      model.compile(

      optimizer=tf.keras.optimizers.Adam(learning_rate=0.001),

      loss=tf.keras.losses.sparse_categorical_crossentropy,

      metrics=[tf.keras.metrics.sparse_categorical_accuracy]

      )大连人流医院哪家好 http://www.dlwkw.com/

      tf.keras.Model.compile 接受 3 个重要的参数:

      oplimizer :优化器,可从 tf.keras.optimizers 中选择;

      loss :损失函数,可从 tf.keras.losses 中选择;

      metrics :评估指标,可从 tf.keras.metrics 中选择。

      接下来,可以使用 tf.keras.Model 的 fit 方法训练模型:

      model.fit(data_loader.train_data, data_loader.train_label, epochs=num_epochs, batch_size=batch_size)

      tf.keras.Model.fit 接受 5 个重要的参数:

      x :训练数据;

      y :目标数据(数据标签);

      epochs :将训练数据迭代多少遍;

      batch_size :批次的大小;

      validation_data :验证数据,可用于在训练过程中监控模型的性能。

      Keras 支持使用 tf.data.Dataset 进行训练,详见 tf.data 。

      最后,使用 tf.keras.Model.evaluate 评估训练效果,提供测试数据及标签即可:

      print(model.evaluate(data_loader.test_data, data_loader.test_label))

  • 相关阅读:
    Poj 2104 K-th Number(主席树&&整体二分)
    Bzoj 3262: 陌上花开(CDQ分治)
    Bzoj 2683: 简单题(CDQ分治)
    ZOJ2314 Reactor Cooling(无源汇上下界可行流)
    Cogs 12. 运输问题2(有上下界的有源汇最大流)
    Cogs 461. [网络流24题] 餐巾(费用流)
    Codevs 1227 方格取数 2(费用流)
    Cogs 13. 运输问题4(费用流)
    Poj 2195 Going Home(费用流)
    开学第二测
  • 原文地址:https://www.cnblogs.com/djw12333/p/14481414.html
Copyright © 2011-2022 走看看