update generate_model_size_report.py to support different cases

SECQUOIA · Aug 18, 2024 · a433ab3 · a433ab3
1 parent b37d6c9
commit a433ab3
Showing 1 changed file with 40 additions and 22 deletions.
diff --git a/generate_model_size_report.py b/generate_model_size_report.py
@@ -5,35 +5,53 @@
 
 
 if __name__ == "__main__":
-    instance_list = [
-        "batch_processing",
-        "biofuel",
-        "cstr",
-        "disease_model",
-        "gdp_col",
-        "hda",
-        "jobshop",
+    instance_dict = {
+        "batch_processing": [],
+        "biofuel": [],
+        "cstr": [],
+        "disease_model": [],
+        "gdp_col": [],
+        "hda": [],
+        "jobshop": [],
         # "kaibel", # next step
-        "med_term_purchasing",
-        "methanol",
-        # "mod_hens", # next step
-        "modprodnet",
-        "positioning",
-        "spectralog",
-        "stranded_gas",
-        "syngas",
-    ]
+        "med_term_purchasing": [],
+        "methanol": [],
+        "mod_hens": [
+            "conventional",
+            "single_module_integer",
+            "multiple_module_integer",
+            "mixed_integer",
+            "single_module_discrete",
+            "multiple_module_discrete",
+            "mixed_discrete",
+        ],
+        "modprodnet": [],
+        "positioning": [],
+        "spectralog": [],
+        "stranded_gas": [],
+        "syngas": [],
+    }
     current_time = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
     timelimit = 600
 
-    for instance in instance_list:
+    for instance, cases in instance_dict.items():
         print("Generating model size report: " + instance)
 
-        model = import_module("gdplib." + instance).build_model()
-        report = build_model_size_report(model)
-        report_df = pd.DataFrame(report.overall, index=[0]).T
+        if cases == []:
+            model = import_module("gdplib." + instance).build_model()
+            report = build_model_size_report(model)
+            report_df = pd.DataFrame(report.overall, index=[0]).T
+
+            report_df.columns = ["Number"]
+        else:
+            report_df = pd.DataFrame()
+            for case in cases:
+                model = import_module("gdplib." + instance).build_model(case)
+                case_report = build_model_size_report(model)
+                temp_df = pd.DataFrame(case_report.overall, index=[0]).T
+                temp_df.columns = [case]
+                report_df = pd.concat([report_df, temp_df], axis=1)
 
-        report_df.columns = ["Number"]
         report_df.index = [
             "Variables",
             "Binary variables",