x

train.py

@@ -28,7 +28,7 @@ parser = argparse.ArgumentParser(description="Train and evaluate HoHo model with
 parser.add_argument('--vertex_threshold', type=float, default=0.32, 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=100, help='Maximum number of samples to process.')
+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.')