import tensorflow as tf import IPython.display as display import numpy as np import matplotlib.pyplot as plt import matplotlib as mpl mpl.rcParams['figure.figsize'] = (12,