Open In App
Related Articles

SciPy – Integration of a Differential Equation for Curve Fit

Improve Article
Save Article
Like Article

In Machine Learning, often what we do is gather data, visualize it, then fit a curve in the graph and then predict certain parameters based on the curve fit. If we have some theoretical data we can use curve fitting from the verified dataset to extract the equation and verify it. So to find the equation of a curve of any order be it linear, quadratic or polynomial, we use Differential Equations and then integrating that equation we can get the curve fit.
In Python SciPy, this process can be done easily for solving the differential equation by mathematically integrating it using odeint(). The odeint(model, y0, t) can be used to solve any order differential equation by taking three or more parameters. 

Parameters : 
model– the differential equation  
y0– Initial value of Y 
t– the time space for which we want the curve(basically the range of x) 

Let’s illustrate this with an example: 

Code: To solve the equation to get y = x – 1 + 2 (e^-x) as the solution 


import numpy as np
from scipy.integrate import odeint
import matplotlib.pyplot as plt
from scipy.integrate import odeint
# Define a function which calculates the derivative by making dy/dx as
# the subject of formula in the given above equation
def dy_dx(y, x):
    return x - y
#Here we define the graph limits for x
xs = np.linspace(-5,5,100)
#Initially y=1 given
y0 = 1.0
# This scipy.integrate.odeint() integrates the given equation by taking
# 3 parameters odeint(model,y0,t)
# model- the differential equation function
# y0-the initial value of y
# t- the timepoints for which we need to plot the curve
ys = odeint(dy_dx, y0, xs)
ys = np.array(ys).flatten()
# By using the matplotlib.pyplot library we plot the curve after integration
plt.rcParams.update({'font.size': 14})  # increase the font size
plt.ylabel("Y- values")
plt.title('Curve of -> y=x-1+2*(e^-x)')
plt.plot(xs, ys,color='green')

This is the graph generated by using the scipy.integrate.odeint( ) which can be seen below and further be used for curve fitting – for analyzing the data in Machine learning. 

The Graph generated after integrating the differential equation


Whether you're preparing for your first job interview or aiming to upskill in this ever-evolving tech landscape, GeeksforGeeks Courses are your key to success. We provide top-quality content at affordable prices, all geared towards accelerating your growth in a time-bound manner. Join the millions we've already empowered, and we're here to do the same for you. Don't miss out - check it out now!

Last Updated : 23 Jan, 2022
Like Article
Save Article
Similar Reads
Complete Tutorials