Separated model training in Snakefile by using wildcards.

Snakefile

@@ -66,17 +66,16 @@ rule train_model:
     input:
         DIR+'/input_data.npy',
         DIR+'/output_data.npy'
-    params:
-        models = MODELS
+    # params:
+    #     # parameter = MODELS[{model}],
+    #     lambda wildcards: [MODELS[wildcards.model]]
     log:
-        DIR+'/log/train_model.log'
+        DIR+'/log/train_model_{model}.log'
     output:
-        expand(DIR+'/trained models/model__{model}.pt', model=MODELS),
-        expand(DIR+'/trained models/loss__{model}.pt', model=MODELS)
+        DIR+'/trained models/model__{model}.pt',
+        DIR+'/trained models/loss__{model}.pt'
     run:
-        for model in params.models:
-            print(model)
-            trainer= ANN_Training.ModelTrainer({'model_name': model, 'dir': DIR,
-                                            'model_dir': DIR, **params.models[model]})
+        trainer= ANN_Training.ModelTrainer({'model_name': wildcards.model, 'dir': DIR,
+                                        'model_dir': DIR, **MODELS[wildcards.model]})
         trainer.epoch_training()
         trainer.save_model()
\ No newline at end of file