From dc7ba04b4308a8a55d8a471a2560bda8ac2746e5 Mon Sep 17 00:00:00 2001 From: Kaushik Sinha Date: Mon, 19 Sep 2022 13:13:06 -0400 Subject: [PATCH] Update utils.py Fix plot_function by adding a steps parameter to torch.linspace. This fixes the issue referenced in https://github.com/fastai/fastbook/issues/399 (__init__.py deprecation warning for linspace usage #399) --- utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils.py b/utils.py index 64cc2cf..9554042 100644 --- a/utils.py +++ b/utils.py @@ -69,8 +69,8 @@ def search_images_ddg(key,max_n=200): pass -def plot_function(f, tx=None, ty=None, title=None, min=-2, max=2, figsize=(6,4)): - x = torch.linspace(min,max) +def plot_function(f, tx=None, ty=None, title=None, min=-2, max=2, steps=100, figsize=(6,4)): + x = torch.linspace(min,max,steps) fig,ax = plt.subplots(figsize=figsize) ax.plot(x,f(x)) if tx is not None: ax.set_xlabel(tx)