I was looking at this example from ml5.js. I am trying to build a small script with this where I can show users the effects of changing batchSizes or number of epochs. Howev