Implementing a One-Layer Neural Network
We have all the tools to implement a neural network that operates on real data. We will create a neural network with one layer that operates on the Iris dataset.
In this section, we will implement a neural network with one hidden layer. It will be important to understand that a fully connected neural network is based mostly on matrix multiplication. As such, the dimensions of the data and matrix are very important to get lined up correctly.
Since this is a regression problem, we will use the mean squared error as the loss function.
How to do it…
- To create the computational graph, we'll start by loading the necessary libraries:
import matplotlib.pyplot as plt import numpy as np import tensorflow as ...