script.js处理训练和推理相关过程代码,而且还包括训练和推理的可视化
import {
MnistData} from './data.js';
var canvas, ctx, saveButton, clearButton;
var pos = {
x:0, y:0};
var rawImage;
var model;
function getModel() {
model = tf.sequential();
model.add(tf.layers.conv2d({
inputShape: [28, 28, 1], kernelSize: 3, filters: 8, activation: 'relu'}));
model.add(tf.layers.maxPooling2d({
poolSize: [2, 2]}));
model.add(tf.layers.conv2d({
filters: 16, kernelSize: 3, activation: 'relu'}));
model.add(tf.layers.maxPooling2d({
poolSize: [2, 2]}));
model.add(tf.layers.flatten());
model.add(tf.layers