Code to build a model, The issue I am having is when I attempt to load model and implement to test dataset, I get the error:
learning_rate=0.001 epochs = 10