Machine Learning, TensorFlow, JavaScript, Javascript Development, Tensorflowjs.
'sigmoid', inputDim: 2, })); model.add(tf.layers.dense({ units: 1, useBias: true, activation: 'sigmoid', })); model.compile({ loss: 'meanSquaredError', optimizer:tf.train.adam() }); return model; } Please find below the training, validation and testing loss of the new model with two input features