Update playout.py

Show statistics
Julian-Theis · May 7, 2021 · d7d785f · d7d785f
1 parent 342d638
commit d7d785f
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/util/playout.py b/util/playout.py
@@ -147,6 +147,14 @@ def standard_playout(pn, f_pop, f_train, f_test, xes_train, csv_train, train_siz
         else:
             test.append(trace)
 
+    print()
+    print("Mean variant length size train:", np.mean([len(s) for s in train]))
+    print("Mean variant length size test:", np.mean([len(s) for s in test]))
+    print()
+    print("Max variant length size train:", np.max([len(s) for s in train]))
+    print("Max variant length size test:", np.max([len(s) for s in test]))
+    print()
+
     writeVariantToFile(f_train, train)
     print("Unique Variant Log (Train, " + str(int(train_size*100)) + "%) stored in:", str(f_train))