Adding a control function

Author: Nivedha-Ramesh

create_dataset.py

@@ -6,8 +6,8 @@ class DatasetCreator:
     """
     A class to create a dataset for the box packing problem, including truck dimensions and box parameters.
     """
-    def __init__(self, min_boxes=10, max_boxes=36, min_value=50, max_value=500,
-                 max_truck_dim=(600, 600, 600), min_truck_dim=(50, 50, 50)):
+    def __init__(self, min_boxes=10, max_boxes=100, min_value=50, max_value=500,
+                 max_truck_dim=(1000, 1000, 1000), min_truck_dim=(50, 50, 50)):
         """
         Initializes the DatasetCreator with parameters for box and truck dimension ranges.
 
@@ -59,3 +59,10 @@ def generate_dataset(self):
     def save_to_file(self, dataset, filename='input.json'):
         with open(filename, 'w') as outfile:
             json.dump(dataset, outfile)
+
+
+if __name__ == "__main__":
+    creator = DatasetCreator()
+    dataset = creator.generate_dataset()
+    creator.save_to_file(dataset)
+    print("New dataset has been generated")