reduce max epochs and train size

Author: ghar1821

src/methods/cytovi/config.vsh.yaml

@@ -38,13 +38,13 @@ arguments:
     type: integer
     default: 1
     description: Number of layers.
-  - name: --n_clusters
+  - name: --max_epochs
     type: integer
-    default: 20
-    description: Number of clusters to use for subsampling.
-  - name: --subsample_fraction
+    default: 500
+    description: Number of epochs to train the model.
+  - name: --train_size
     type: double
-    default: 0.5
+    default: 0.7
     description: Fraction of cells to subsample from each cluster for training.
 
 # Resources required to run the component

src/methods/cytovi/script.py

@@ -15,6 +15,8 @@
     "output": "resources_test/task_cyto_batch_integration/mouse_spleen_flow_cytometry_subset/output_cytovi_split2.h5ad",
     "n_hidden": 128,
     "n_layers": 1,
+    "max_epochs": 500,
+    "train_size": 0.7,
 }
 meta = {"name": "cytovi"}
 ## VIASH END
@@ -57,6 +59,8 @@
 start = time.time()
 model.train(
     batch_size=8192,
+    max_epochs=par["max_epochs"],
+    train_size=par["train_size"],
 )
 end = time.time()
 print(f"Training took {end - start:.2f} seconds", flush=True)