Update pipeline_test.py

pipeline_test.py

@@ -28,10 +28,12 @@ sweep_configuration = {
         'N_TILES': {'values': [3]},
         'SLICE_THICKNESS': {'values': [0.25, 0.5, 0.75]},
         'FIND_STEMS_HEIGHT': {'values': [0.5, 0.75, 1.0, 1.5, 2.0]},
-        'FIND_STEMS_THICKNESS': {'values': [0.25, 0.5, 0.75]},
-        'GRAPH_MAXIMUM_CUMULATIVE_GAP': {'values': [1, 2, 3, 4]},
-        'ADD_LEAVES_VOXEL_LENGTH': {'values': [0.25, 0.5, 0.75]},
-        'FIND_STEMS_MIN_POINTS': {'values': [10, 20, 30, 50, 100, 150, 200]}
+        'FIND_STEMS_THICKNESS': {'values': [0.25, 0.5, 0.75]}, # 0.1 - 1.0
+        'GRAPH_MAXIMUM_CUMULATIVE_GAP': {'values': [5, 10, 15, 20]}, # 5 - 20
+        'ADD_LEAVES_VOXEL_LENGTH': {'values': [0.1, 0.25, 0.5, 0.75]}, # 0.1 - 0.5
+        'FIND_STEMS_MIN_POINTS': {'values': [10, 20, 30, 50, 100, 150, 200]}, # 50 - 500
+        'GRAPH_EDGE_LENGTH' : {'values': [0.5, 1.0, 1.5, 2.0]},
+        'ADD_LEAVES_EDGE_LENGTH': {'values': [0.2, 0.5, 0.75, 1.0, 1.5]}
     }
 }
 
@@ -107,4 +109,4 @@ def main():
 sweep_id = wandb.sweep(sweep=sweep_configuration, project="sweep-train", entity="maciej-wielgosz-nibio")
 
 # run the sweep
-wandb.agent(sweep_id, function=main, count=1000)
+wandb.agent(sweep_id, function=main, count=1000)
\ No newline at end of file