diff --git a/machine_learning/logistic_regression.py b/machine_learning/logistic_regression.py index de0cfd54d2c6..eee6c46dda86 100644 --- a/machine_learning/logistic_regression.py +++ b/machine_learning/logistic_regression.py @@ -16,6 +16,10 @@ #importing all the required libraries import numpy as np import matplotlib.pyplot as plt +try: + get_ipython +except NameError: + from IPython import get_ipython get_ipython().run_line_magic('matplotlib', 'inline') from sklearn import datasets