correct bug

modules/toXML.py

@@ -434,6 +434,12 @@ def create_BPMN_id(labels, pool_dict):
     Returns:
         tuple: List of BPMN IDs and updated pool dictionary.
     """
+
+    #change the label to task if it's subProcess
+    for i in range(len(labels)):
+        if labels[i] == list(class_dict.values()).index('subProcess'):
+            labels[i] = list(class_dict.values()).index('task')
+
     BPMN_id = [class_dict[labels[i]] for i in range(len(labels))]
 
     data_counter = 1

modules/train.py

@@ -95,7 +95,7 @@ def prepare_model(dict, opti, learning_rate=0.0003, model_to_load=None, model_ty
     if opti == 'SGD':
         optimizer = SGD(model.parameters(), lr=learning_rate, momentum=0.9, weight_decay=0.0001)
     elif opti == 'Adam':
-        optimizer = AdamW(model.parameters(), lr=learning_rate, weight_decay=0.00056, eps=1e-08, betas=(0.9, 0.999))
+        optimizer = AdamW(model.parameters(), lr=learning_rate, weight_decay=0.001, eps=1e-08, betas=(0.9, 0.999))
     else:
         print('Optimizer not found')