pip

Author: Borda

.actions/assistant.py

@@ -372,7 +372,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} --quiet {pip_args}", "pip list"]
+            cmd += [f"pip install {pip_req} {pip_args}", "pip list"]  # TODO: add ` --quiet`
             # 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

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