pip

Author: Borda

.actions/assistant.py

@@ -322,7 +322,7 @@ def bash_render(folder: str, output_file: str = PATH_SCRIPT_RENDER) -> Optional[
             cmd += [f"cp {meta_file} {pub_meta}", f"cat {pub_meta}", f"git add {pub_meta}"]
         else:
             pip_req, pip_args = AssistantCLI._parse_requirements(folder)
-            cmd += [f"pip install {pip_req} {pip_args}", "pip list"]
+            cmd += [f"pip install {pip_req} --quiet {pip_args}", "pip list"]
             cmd.append(f"# available: {AssistantCLI.DEVICE_ACCELERATOR}\n")
             if AssistantCLI._valid_accelerator(folder):
                 cmd.append(f"python -m papermill {ipynb_file} {pub_ipynb} --kernel python")
@@ -369,7 +369,7 @@ def bash_test(folder: str, output_file: str = PATH_SCRIPT_TEST) -> Optional[str]
         if AssistantCLI._valid_accelerator(folder):
             # and install specific packages
             pip_req, pip_args = AssistantCLI._parse_requirements(folder)
-            cmd += [f"pip install {pip_req} {pip_args}", "pip list"]
+            cmd += [f"pip install {pip_req} --quiet {pip_args}", "pip list"]
             # Export the actual packages used in runtime
             cmd.append(f"meta_file=$(python .actions/assistant.py update-env-details {folder} --base_path .)")
             # show created meta config

requirements/default.txt

@@ -1,4 +1,4 @@
 ipython[notebook]
-pytorch-lightning>=1.3
-torchmetrics>=0.3
+pytorch-lightning>=1.4
+torchmetrics>=0.6
 torch>=1.6, <1.9

templates/titanic/tutorial.py

@@ -39,7 +39,7 @@
     target_fields="Survived",
     train_file=csv_train,
     val_split=0.1,
-    batch_size=8,
+    batch_size=32,
 )
 
 # %% [markdown]
@@ -49,7 +49,7 @@
 model = TabularClassifier.from_data(
     datamodule,
     learning_rate=0.1,
-    optimizer="Adam",
+    optimizer="AdamW",
     n_a=8,
     gamma=0.3,
 )
@@ -81,6 +81,7 @@
 sns.relplot(data=metrics, kind="line")
 plt.gca().set_ylim([0, 1.25])
 plt.gcf().set_size_inches(10, 5)
+plt.grid()
 
 # %% [markdown]
 # ## 4. Generate predictions from a CSV