TensorFlowJS-Model

本文介绍了TensorFlow.js中创建模型的两种方式:tf.sequential和tf.model。tf.sequential用于创建层与层之间简单栈式结构的模型,而tf.model则适用于非栈式结构,当一层的输出不完全传递给下一层时。同时提到了如何从保存的参数中加载模型。
摘要由CSDN通过智能技术生成

模型用来描述Layer之间的拓扑逻辑关系,即一个Layer的输出,以何种形式作为下一个Layer的输入。模型可以被用来训练和预测。模型的状态(包括拓扑逻辑,训练得到的权重)可以从其他的格式里面恢复过来。

有两种创建模型的方式: tf.sequential和tf.model。

tf.sequential

简单的类似栈一样的关系。就是每一层的输入,依赖于上一层的输出(我理解的是:一层的某个Tensor的输出的数目,等于下一层的所有的Tensor的数目,并且每个输出对应下一层的一个Tensor的输入)。
图片引用自:https://medium.freecodecamp.org/get-to-know-tensorflow-js-in-7-minutes-afcd0dfd3d2f
图片引用自:https://medium.freecodecamp.org/get-to-know-tensorflow-js-in-7-minutes-afcd0dfd3d2f

使用tf.sequential();来创建:

	function createDenseModel() {
	  const model = tf.sequential();
	  model.add(tf.layers.flatten({inputShape: [IMAGE_H, IMAGE_W, 1]}));
	  model.add(
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值