Load Model in TFJS

load_model_tfjs.js

const tf = require('@tensorflow/tfjs-node')

const load_model = async () => {
	const model = await tf.loadLayersModel(
		'file:///path/to/directory/tfjs/model.json'
	)
	model.weights.forEach((w) => {
		console.log(w.name, w.shape)
	})

	model.compile({
		optimizer: 'sgd',
		loss: 'categoricalCrossentropy',
		metrics: ['accuracy'],
	})
	return model
}