chris/fastbook
1
0
Fork 0
mirror of https://github.com/fastai/fastbook.git synced 2025-04-04 01:40:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update utils.py

Browse Source 
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)
This commit is contained in:
Kaushik Sinha 2022-09-19 13:13:06 -04:00
parent 2b8b8a2097
commit dc7ba04b43
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils.py
View File

@ -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)