Published
Edited
Apr 6, 2018
9 stars
Insert cell
Insert cell
model.predict(tf.zeros([1, 28, 28, 1])).data()
Insert cell
model = {
const model = tf.sequential();
model.add(tf.layers.conv2d({
inputShape: [28, 28, 1],
kernelSize: 5,
filters: 8,
strides: 1,
activation: "relu",
kernelInitializer: "VarianceScaling"
}));

model.add(tf.layers.maxPooling2d({
poolSize: [2, 2],
strides: [2, 2]
}));

model.add(tf.layers.conv2d({
kernelSize: 5,
filters: 16,
strides: 1,
activation: "relu",
kernelInitializer: "VarianceScaling"
}));

model.add(tf.layers.maxPooling2d({
poolSize: [2, 2],
strides: [2, 2]
}));

model.add(tf.layers.flatten());

model.add(tf.layers.dense({
units: 10,
kernelInitializer: "VarianceScaling",
activation: "softmax"
}));
await model.compile({
optimizer: tf.train.adam(0.0001),
loss: "categoricalCrossentropy",
metrics: ["accuracy"]
});
return model;
}
Insert cell
tf = require("@tensorflow/tfjs@0.8.0/dist/tf.min.js")
Insert cell

Purpose-built for displays of data

Observable is your go-to platform for exploring data and creating expressive data visualizations. Use reactive JavaScript notebooks for prototyping and a collaborative canvas for visual data exploration and dashboard creation.
Learn more