https://codelabs.developers.google.com/codelabs/tfjs-training-classfication/index.html