Load MNIST data and convert to tfjs tensors

load_mnist_data_tfjs.js

const load_data = require('./data_loader')

const data = load_data()

const X = data.map((elem) => {
	const key = Object.keys(elem)[0]
	return elem[key].map((val) => val / 255)
})
console.log(X[0])
const arr = Array.apply(null, Array(10)).map(() => 0)
const y = data.map((elem) => {
	const key = parseInt(Object.keys(elem)[0])
	const copy_arr = Object.assign([], arr)
	for (let index = 0; index < copy_arr.length; index++) {
		if (index === key) {
			copy_arr[index] = 1
		}
	}
	return copy_arr
})
console.log(y[0],y[1])

console.log('X:', X.length)
console.log('y:', y.length)
let X_tensor = tf.tensor2d(X, [5000, 784])
let y_tensor = tf.tensor2d(y, [5000, 10])