Linear Regression Using Tensorflow


Predictive analysis makes heavy use of linear regression, a key idea in machine learning and data analysis. The top open-source machine learning framework TensorFlow offers powerful tools for putting linear regression models into practise. By using concrete examples, this article will guide you through the specifics of linear regression in the context of TensorFlow.

Understanding Linear Regression

By fitting the data to a linear equation, the predictive statistical technique of linear regression seeks to simulate the connection between a dependent variable and one or more independent variables. In essence, it uses historical data to anticipate the result for a specific input.

TensorFlow and Linear Regression

To build and complete tasks involving linear regression, TensorFlow provides many tools and libraries. Utilising the power of deep learning, TensorFlow enables you to create linear regression models that can handle large datasets effectively.

Creating Linear Regression Models with TensorFlow: Practical Examples

Make sure TensorFlow is set up in your Python environment before we start. Use the pip command below to install if not −

pip install tensorflow

Example 1: Simple Linear Regression with TensorFlow

The example that follows shows how to use TensorFlow to build a straightforward linear regression model:

import numpy as np
import tensorflow as tf

# Create some example data
X_train = np.linspace(0, 10, 100)
y_train = 2*X_train + np.random.randn(*X_train.shape)*0.33

# Define model parameters
learning_rate = 0.01
training_epochs = 100

# Define placeholders for inputs and outputs
X = tf.placeholder(tf.float32)
Y = tf.placeholder(tf.float32)

# Define variables to be learned
W = tf.Variable(np.random.randn(), name="weight")
b = tf.Variable(np.random.randn(), name="bias")

# Define the linear regression model
pred = tf.add(tf.multiply(X, W), b)

# Define the cost function (Mean Squared Error)
cost = tf.reduce_sum(tf.pow(pred-Y, 2)) / (2 * X_train.shape[0])

# Define the optimizer (Gradient Descent)
optimizer = tf.train.GradientDescentOptimizer(learning_rate).minimize(cost)

# Initialize all the variables
init = tf.global_variables_initializer()

# Start the TensorFlow session and run the model
with tf.Session() as sess:

   # Training phase
   for epoch in range(training_epochs):
      for (x, y) in zip(X_train, y_train):, feed_dict={X: x, Y: y})

   # Print the final learned parameters
   print("Optimized weight:",
   print("Optimized bias:",

In this example, we'll use a straightforward linear regression model to discover how X_train and y_train are related. By using gradient descent optimisation, the model parameters are learned.

Example 2: Multiple Linear Regression with TensorFlow

This example shows how to use TensorFlow to build a multiple linear regression model −

import numpy as np
import tensorflow as tf

# Create some example data
X_train = np.array([[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]], dtype=np.float32)
y_train = np.array([[2], [4], [6], [8], [10]], dtype=np.float32)

# Define model parameters
learning_rate = 0.01
training_epochs = 1000

# Define the linear regression model
model = tf.keras.models.Sequential([
  tf.keras.layers.Dense(1, input_shape=(2,), activation='linear')])
# Define the optimizer and compile the model
optimizer = tf.keras.optimizers.SGD(learning_rate)
model.compile(optimizer=optimizer, loss='mse')

#Train the model, y_train, epochs=training_epochs)

#Print the learned parameters
weights, bias = model.layers[0].get_weights()
print("Optimized weights:", weights)
print("Optimized bias:", bias)

In this instance, a multiple linear regression model was produced using the Keras API of TensorFlow. Two input characteristics, a linear activation function, and a single dense layer make up the model. Mean Squared Error (MSE) has been employed as the loss function and Stochastic Gradient Descent (SGD) as the optimizer.

Harnessing the Power of Linear Regression in TensorFlow

TensorFlow's linear regression models are robust and adaptable. They provide a quick, efficient method for using data to analyse and make predictions. While multiple linear regression can be implemented using TensorFlow and handle more input features and hence more complex scenarios, simple linear regression can still be quite useful in some situations.

TensorFlow not only offers tools for building linear regression models, but also methods for assessing those models, including multiple measures for loss and accuracy. Furthermore, you can easily preprocess data, postprocess findings, and visualise the performance of the model thanks to TensorFlow's compatibility with existing data manipulation and visualisation packages.


Learning how to use TensorFlow's linear regression models is a big step towards mastering this potent machine learning toolkit. You can open up a wide range of opportunities in data analysis and predictive modelling by learning how to define, train, and optimise these models. The abilities you've gained in this course provide a great foundation for approaching various jobs, whether you're projecting sales, predicting home prices, or spotting trends. So dive in, try things out, and leverage TensorFlow's strength to uncover the insights buried in your data!

Updated on: 18-Jul-2023


Kickstart Your Career

Get certified by completing the course

Get Started