跳到主要内容

tf.model

2025年02月10日
柏拉文
越努力,越幸运

一、认识


二、语法


const input = tf.input({ shape: [4] });
const dense1 = tf.layers.dense({ units: 8, activation: 'relu' }).apply(input);
const output = tf.layers.dense({ units: 1, activation: 'sigmoid' }).apply(dense1);
const model = tf.model({ inputs: input, outputs: output });