Building Your First Neural Network with TensorFlow

In this lesson, we will create a simple feedforward neural network using TensorFlow. TensorFlow is a popular framework for building and training machine learning models.

Step 1: Setting Up TensorFlow

First, ensure that TensorFlow is installed. If not, you can install it using:

pip install tensorflow

We’ll also import the required libraries for this lesson.

import tensorflow as tf
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense
import numpy as np

Step 2: Preparing the Dataset

For this example, we’ll use a simple dataset where the input is a single feature, and the output is its double:

# Generate sample data
X = np.array([[1], [2], [3], [4], [5]], dtype=float)
y = np.array([[2], [4], [6], [8], [10]], dtype=float)

Step 3: Creating the Model

We’ll create a simple model with one input layer, one hidden layer, and one output layer. The hidden layer will use the ReLU activation function.

# Create the model
model = Sequential([
    Dense(10, activation='relu', input_shape=(1,)),  # Hidden layer with 10 neurons
    Dense(1)  # Output layer with 1 neuron
])

Step 4: Compiling the Model

We compile the model with:

• Loss Function: Mean Squared Error for regression tasks.
• Optimizer: Stochastic Gradient Descent (SGD).

# Compile the model
model.compile(optimizer='sgd', loss='mean_squared_error')

Step 5: Training the Model

We train the model using the training data for 500 epochs. During training, the model learns to minimize the loss.

# Train the model
model.fit(X, y, epochs=500, verbose=1)

Step 6: Making Predictions

After training, we can use the model to make predictions on new data:

# Predict on new data
print(model.predict([6]))

Step 7: Evaluating the Model

We evaluate the model’s performance on the training data to ensure it has learned the pattern correctly:

# Evaluate the model
loss = model.evaluate(X, y)
print(f"Loss: {loss}")

Real-World Applications

Feedforward neural networks built using TensorFlow can be applied to:

• Predictive Analytics: Forecasting sales and trends.
• Classification: Spam detection, fraud detection.
• Regression: Price prediction, demand forecasting.

Summary and Next Steps

In this lesson, we:

• Built and trained a simple neural network using TensorFlow.
• Learned about the key steps: dataset preparation, model creation, compilation, and training.
• Made predictions and evaluated the model’s performance.

Next, we’ll explore how neural networks are extended to more advanced architectures like convolutional neural networks (CNNs) for image processing tasks.