瀏覽代碼

fix the tutorial

sunyanfang01 5 年之前
父節點
當前提交
0d54ff26d5
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      tutorials/interpret/interpret.py

+ 2 - 2
tutorials/interpret/interpret.py

@@ -34,12 +34,12 @@ model = pdx.load_model('mini_imagenet_veg_mobilenetv2')
 save_dir = 'interpret_results'
 save_dir = 'interpret_results'
 if not osp.exists(save_dir):
 if not osp.exists(save_dir):
     os.makedirs(save_dir)
     os.makedirs(save_dir)
-visualize('mini_imagenet_veg/mushroom/n07734744_1106.JPEG', 
+pdx.interpret.visualize('mini_imagenet_veg/mushroom/n07734744_1106.JPEG', 
           model,
           model,
           test_dataset, 
           test_dataset, 
           algo='lime',
           algo='lime',
           save_dir=save_dir)
           save_dir=save_dir)
-visualize('mini_imagenet_veg/mushroom/n07734744_1106.JPEG', 
+pdx.interpret.visualize('mini_imagenet_veg/mushroom/n07734744_1106.JPEG', 
           model, 
           model, 
           test_dataset, 
           test_dataset, 
           algo='normlime',
           algo='normlime',