From the course: AWS Certified Machine Learning - Specialty (MLS-C01) Cert Prep: 3 Modeling

Unlock this course with a free trial

Join today to access over 23,100 courses taught by industry experts.

Neural network architecture

Neural network architecture

- [Instructor] A great way to learn more about the intuition behind neural networks is to play with the playground.tensorflow.org tool. And here we see that we've got a data set selected that has two distinct classes, and we have a couple features here. And we have some neurons here in one layer and we have another hidden layer with another couple of neurons here. And we can see the test loss and the training loss. So if I wanted to start training this model, you can see here this is the learning rate and eventually it starts to figure out the difference between these two classes. But I do have a little bit of overfitting here because of this test loss. So one of the things that I could do is I could try to apply some other hyper parameter tuning so I could change the learning rate to potentially find the solution a little bit quicker. And I also could tweak the regularization to prevent overfitting. But first, let's…

Contents