Need to test the materials

leschultz · Apr 18, 2024 · 06a4990 · 06a4990
1 parent 72f2810
commit 06a4990
Show file tree

Hide file tree

Showing 38 changed files with 43 additions and 5,191 deletions.
diff --git a/examples/materials/asr/fit.py b/examples/materials/asr/fit.py
diff --git a/examples/materials/asr/run.sh b/examples/materials/asr/run.sh
diff --git a/examples/materials/asr/submit.sh b/examples/materials/asr/submit.sh
diff --git a/examples/materials/combined/fit.py b/examples/materials/combined/fit.py
@@ -13,7 +13,14 @@ def main():
 
     # Data
     X, y = datasets.load(tasks)
-    data = datasets.splitter(X, y, tasks, train_size=1)
+    data = datasets.splitter(
+                             X, 
+                             y,
+                             tasks,
+                             train_size=0.8,
+                             val_size=0.1,
+                             test_size=0.1,
+                             )
 
     for k, v in data.items():
         data[k]['scaler'] = StandardScaler()

diff --git a/examples/synthetic/fit.py b/examples/synthetic/fit.py
@@ -8,12 +8,19 @@ def main():
     save_dir = 'outputs'
     lr = 1e-4
     batch_size = 32
-    n_epochs = 100
+    n_epochs = 1000
     tasks = ['data1', 'data2', 'data3']
 
     # Data
     X, y = datasets.toy()
-    data = datasets.splitter(X, y, tasks, train_size=1)
+    data = datasets.splitter(
+                             X,
+                             y,
+                             tasks,
+                             train_size=0.8,
+                             val_size=0.1,
+                             test_size=0.1,
+                             )
 
     for k, v in data.items():
         data[k]['scaler'] = StandardScaler()

diff --git a/examples/synthetic/outputs/data1/X_train.csv b/examples/synthetic/outputs/data1/X_train.csv
diff --git a/examples/synthetic/outputs/data1/loss.pkl b/examples/synthetic/outputs/data1/loss.pkl
diff --git a/examples/synthetic/outputs/data1/loss_vs_epoch/train.png.json b/examples/synthetic/outputs/data1/loss_vs_epoch/train.png.json
diff --git a/examples/synthetic/outputs/data1/loss_vs_epoch/train.png.png b/examples/synthetic/outputs/data1/loss_vs_epoch/train.png.png
diff --git a/examples/synthetic/outputs/data1/parity/train.png.json b/examples/synthetic/outputs/data1/parity/train.png.json
diff --git a/examples/synthetic/outputs/data1/parity/train.png.png b/examples/synthetic/outputs/data1/parity/train.png.png
diff --git a/examples/synthetic/outputs/data1/parity/train.png_legend.png b/examples/synthetic/outputs/data1/parity/train.png_legend.png
diff --git a/examples/synthetic/outputs/data1/scaler.joblib b/examples/synthetic/outputs/data1/scaler.joblib