dataset_creation package

• create_dataset module
  • check_gpu()
  • create_gprmax_input_files()
  • run_simulations()
• geom2bscan module
  • build_network()
  • check_shapes_equal()
  • filter_PML_bug_bscans()
  • filter_initial_wave()
  • load_dataset()
  • predict()
  • predict_batch()
  • predict_batch_wide()
  • preprocess_data()
  • save_predictions()
  • split_dataset()
  • train()
• inputfile module
  • InputFile
    • InputFile.close()
    • InputFile.sample_randomized_metadata()
    • InputFile.write_ballast()
    • InputFile.write_box_material()
    • InputFile.write_command()
    • InputFile.write_fractal_box_material()
    • InputFile.write_full_track()
    • InputFile.write_general_commands()
    • InputFile.write_line()
    • InputFile.write_materials()
    • InputFile.write_pss()
    • InputFile.write_rails()
    • InputFile.write_randomized()
    • InputFile.write_save_geometry()
    • InputFile.write_sleepers()
    • InputFile.write_snapshots()
    • InputFile.write_source_receiver()
• ballast_simulation module
  • BallastSimulation
    • BallastSimulation.get_clean_ballast_radii_distrib()
    • BallastSimulation.get_fouled_ballast_radii_distrib()
    • BallastSimulation.random_sequential_adsorption()
    • BallastSimulation.run()
    • BallastSimulation.sample_radii_distribution()
• convert_to_np module
  • convert_geometry_to_np()
  • convert_snapshots_to_np()
  • extract_snapshot_fields_numpy()
• configuration module
  • GprMaxConfig
    • GprMaxConfig.antenna_sleeper_distance
    • GprMaxConfig.create_views
    • GprMaxConfig.domain_size
    • GprMaxConfig.fouling_beta_params
    • GprMaxConfig.fouling_box_threshold
    • GprMaxConfig.fractal_dimension
    • GprMaxConfig.general_deterioration_beta_params
    • GprMaxConfig.general_water_content_beta_params
    • GprMaxConfig.generate_input
    • GprMaxConfig.geometry_only
    • GprMaxConfig.input_dir
    • GprMaxConfig.layer_roughness
    • GprMaxConfig.layer_sizes
    • GprMaxConfig.layer_sizes_beta_params
    • GprMaxConfig.layer_water_sampling_std
    • GprMaxConfig.materials
    • GprMaxConfig.n_ascans
    • GprMaxConfig.n_samples
    • GprMaxConfig.output_dir
    • GprMaxConfig.pep_soil_number
    • GprMaxConfig.receiver_position
    • GprMaxConfig.run_simulations
    • GprMaxConfig.seed
    • GprMaxConfig.sleepers_material_probabilities
    • GprMaxConfig.sleepers_separation
    • GprMaxConfig.sleepers_sizes
    • GprMaxConfig.snapshot_times
    • GprMaxConfig.source_amplitude
    • GprMaxConfig.source_central_frequency
    • GprMaxConfig.source_position
    • GprMaxConfig.source_waveform
    • GprMaxConfig.spatial_resolution
    • GprMaxConfig.step_size
    • GprMaxConfig.time_window
    • GprMaxConfig.tmp_dir
    • GprMaxConfig.track_configuration_probabilities
    • GprMaxConfig.water_infiltration_sampling_std
    • GprMaxConfig.water_infiltration_threshold
• statistics module
  • DatasetStats
    • DatasetStats.write_metadata_files()
  • Metadata
    • Metadata.ballast_simulation_seed
    • Metadata.fouling_level
    • Metadata.fouling_material
    • Metadata.general_deterioration
    • Metadata.general_water_content
    • Metadata.is_fouled
    • Metadata.layer_sizes
    • Metadata.layer_water_ranges
    • Metadata.pss_material
    • Metadata.seed
    • Metadata.sleeper_positions
    • Metadata.sleepers_material
    • Metadata.subsoil_material
    • Metadata.track_type
    • Metadata.water_infiltrations