Update to the hyperparameters, so they are the winning solution :)

script.py

@@ -80,7 +80,7 @@ if __name__ == "__main__":
 
     voxel_model = None
 
-    config = {'vertex_threshold': 0.60, 'edge_threshold': 0.68, 'only_predicted_connections': True}
+    config = {'vertex_threshold': 0.59, 'edge_threshold': 0.65, 'only_predicted_connections': True}
 
     print('------------ Now you can do your solution ---------------')
     solution = []

train.py

@@ -35,8 +35,8 @@ import time
 
 # --- Argument Parsing ---
 parser = argparse.ArgumentParser(description="Train and evaluate HoHo model with custom config.")
-parser.add_argument('--vertex_threshold', type=float, default=0.72, help='Vertex threshold for prediction.')
-parser.add_argument('--edge_threshold', type=float, default=0.72, help='Edge threshold for prediction.')
+parser.add_argument('--vertex_threshold', type=float, default=0.59, help='Vertex threshold for prediction.')
+parser.add_argument('--edge_threshold', type=float, default=0.65, help='Edge threshold for prediction.')
 parser.add_argument('--only_predicted_connections', type=lambda x: (str(x).lower() == 'true'), default=True, help='Use only predicted connections (True/False).')
 parser.add_argument('--max_samples', type=int, default=50000, help='Maximum number of samples to process.')
 parser.add_argument('--results_dir', type=str, default="results", help='Directory to save result files.')