custom params

models/fm4m.py

@@ -383,7 +383,7 @@ def single_modal(model,dataset=None, downstream_model=None, params=None, x_train
         print(f'x_batch_test Nan index: {nan_indices}')
         print(f'x_batch_test shape: {x_batch_test.shape}, y_batch_test len: {len(y_batch_test)}')
 
-    print(f" Calculating ROC AUC Score ...")
+    print(f"Model selected: {downstream_model} - Calculating ROC AUC Score ...")
 
     if downstream_model == "XGBClassifier":
         if params == None:
@@ -717,7 +717,7 @@ def multi_modal(model_list,dataset=None, downstream_model=None,params=None, x_tr
 
         print("Generating latent plots : Done")
 
-    print(f" Calculating ROC AUC Score ...")
+    print(f"Model selected: {downstream_model} - Calculating ROC AUC Score ...")
 
 
     if downstream_model == "XGBClassifier":