Tensorflow.js tf.model()用法及代码示例

Tensorflow.js是由Google开发的开源库，用于在浏览器或节点环境中运行机器学习模型以及深度学习神经网络。

tf.model() 函数用于创建一个模型，其中包含以输入和输出参数形式提供的层和层。

用法:

tf.model( args ) 

这里的参数是，

• Inputs:模型的输入。它可以是对象或对象列表。
• Outputs:模型的输出。
• Name:型号名称。

范例1：在这个例子中，我们将在 tf.model() 函数的帮助下创建一个模型，输入大小为 4，然后是 2 个具有激活函数 relu 和 softmax 的密集层，并使用 model.predict() 函数进行预测。

// Create input of size 4
var input = tf.input({shape:[4]});
  
// Dense layer 1 with relu activation 
var dLayer1 = tf.layers.dense({units:12,activation:'relu'});
  
// Dense layer 1 with softmax activation
var dLayer2 = tf.layers.dense({units:7, activation:'softmax'});
  
var output = dLayer2.apply(dLayer1.apply(input));
  
// Model function
var model = tf.model({inputs:input, outputs:output});
  
// Prediction
model.predict(tf.ones([2,4])).print();

输出：

Tensor
[[0.309215, 0.0659644, 0.122767, 0.1150663, 0.1592857, 0.1232278, 0.1044738],
[0.309215, 0.0659644, 0.122767, 0.1150663, 0.1592857, 0.1232278, 0.1044738]]

范例2：在这个例子中，我们将创建一个输入数组大小为 2 的模型，并使用 model.summery() 函数生成摘要。summery() 函数也使用 apply() 和 concatenate() 函数。

// Input 1
var inp1 = tf.input({shape:[12]});
  
// Input 2
var inp2 = tf.input({shape:[24]});
  
// Apply input one to the first dense layer
// uisng apply() function
var denseLayer1 = tf.layers.dense({units:4}).apply(inp1);
  
// Apply input two to second dense layer
var denseLayer2 = tf.layers.dense({units:8}).apply(inp2);
  
// Concatinate both dense layer using concatinate() function
var concatAll = tf.layers.concatenate()
        .apply([denseLayer1,denseLayer2]);
  
var output =tf.layers.dense({units:8, 
        activation:'softmax'}).apply(concatAll);
       
// Create model
var model = tf.model({inputs:[inp1, inp2], outputs:output});
  
// Generate summery for model
model.summary();

输出：

__________________________________________________________________________________________________
Layer (type)                    Output shape         Param #     Receives inputs                  
==================================================================================================
input7 (InputLayer)             [null,12]            0                                            
__________________________________________________________________________________________________
input8 (InputLayer)             [null,24]            0                                            
__________________________________________________________________________________________________
dense_Dense10 (Dense)           [null,4]             52          input7[0][0]                     
__________________________________________________________________________________________________
dense_Dense11 (Dense)           [null,8]             200         input8[0][0]                     
__________________________________________________________________________________________________
concatenate_Concatenate4 (Conca [null,12]            0           dense_Dense10[0][0]              
                                                                 dense_Dense11[0][0]              
__________________________________________________________________________________________________
dense_Dense12 (Dense)           [null,8]             104         concatenate_Concatenate4[0][0]   
==================================================================================================
Total params:356
Trainable params:356
Non-trainable params:0
__________________________________________________________________________________________________

参考： https://js.tensorflow.org/api/latest/#model