--- a
+++ b/AICare-baselines/configs/config.yaml
@@ -0,0 +1,6 @@
+stage: tune # tune / train (k-fold) / test
+batch_size: 81920 # make sure the default value contains the whole dataset
+epochs: 50
+patience: 10
+learning_rate: 1e-3
+output_dim: 1