snowstorm_dataset

Snowstorm dataset module hosted on the IceCube Collaboration servers.

class graphnet.datasets.snowstorm_dataset.SnowStormDataset(run_ids, graph_definition, download_dir, truth, features, train_dataloader_kwargs, validation_dataloader_kwargs, test_dataloader_kwargs)[source]

Bases: IceCubeHostedDataset

IceCube SnowStorm simulation dataset.

More information can be found at https://wiki.icecube.wisc.edu/index.php/SnowStorm_MC#File_Locations This is a IceCube Collaboration simulation dataset. Requires a username and password.

Construct SnowStormDataset.

Parameters:

  • run_ids (List[int]) – List of RunIDs to include.

  • graph_definition (GraphDefinition) – Method that defines the data representation.

  • download_dir (str) – Directory to download dataset to.

  • truth (Optional) – List of event-level truth to include. Will include all available information if not given.

  • features (Optional) – List of input features from pulsemap to use. If not given, all available features will be used.

  • train_dataloader_kwargs (Optional) – Arguments for the training DataLoader. Default None.

  • validation_dataloader_kwargs (Optional) – Arguments for the validation DataLoader, Default None.

  • test_dataloader_kwargs (Optional) – Arguments for the test DataLoader. Default None.