fix: update forest_example notebook

dreamquark-ai · May 10, 2020 · 8092324 · 8092324
1 parent 1642909
commit 8092324
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/forest_example.ipynb b/forest_example.ipynb
@@ -42,7 +42,7 @@
    "source": [
     "url = \"https://archive.ics.uci.edu/ml/machine-learning-databases/covtype/covtype.data.gz\"\n",
     "dataset_name = 'forest-cover-type'\n",
-    "tmp_out = Path(os.getcwd().rsplit(\"/\",  1)[0]+'/data/'+dataset_name+'.gz')\n",
+    "tmp_out = Path('./data/'+dataset_name+'.gz')\n",
     "out = Path(os.getcwd()+'/data/'+dataset_name+'.csv')"
    ]
   },
@@ -199,7 +199,7 @@
     "    optimizer_fn=torch.optim.Adam,\n",
     "    scheduler_params = {\"gamma\": 0.95,\n",
     "                     \"step_size\": 20},\n",
-    "    scheduler_fn=torch.optim.lr_scheduler.StepLR, saving_path=\"./\", epsilon=1e-15\n",
+    "    scheduler_fn=torch.optim.lr_scheduler.StepLR, epsilon=1e-15\n",
     ")"
    ]
   },