load_dataset

Module Contents

load_dataset.load_messier_catalog_images(path, img_size=None, disable_tqdm=False)

Data loader for Messier catalog images. The images are available in messier-catalog-images repository of MiraPy organisation.

Parameters
  • path – String. Directory path.

  • img_size – Final dimensions of the image.

  • disable_tqdm – Boolean. Set True to disable progress bar.

Returns

Array of images.

load_dataset.prepare_messier_catalog_images(images, psf, sigma)

Function to apply convolution and add noise from poisson distribution on an array of images.

Parameters
  • images – Array of images.

  • psf – Point Spread Function (PSF).

  • sigma – Float. VStandard deviation.

Returns

Original image arrays and convolved image arrays.

load_dataset.load_xray_binary_data(path, standard_scaler=True)

Loads X Ray Binary dataset from directory.

Parameters
  • path – Path to the directory.

  • standard_scaler – Bool. Standardize data or not.

Returns

Dataset and Class labels.

load_dataset.load_atlas_star_data(path, standard_scaler=True, feat_list=None)

Loads ATLAS variable star dataset from directory.

Parameters
  • path – Path to the directory.

  • standard_scaler – Bool. Standardize data or not.

  • feat_list – List of features to include in dataset.

Returns

Dataset and Class labels.

load_dataset.load_ogle_dataset(path, classes, time_len=50, pad=False)

Loads OGLE variable star time series data from directory.

Parameters
  • path – Path to the directory.

  • classes – Classes to include in dataset.

  • time_len – Length of time series data.

  • pad – Bool. Pad zeroes or not.

Returns

Dataset and Class labels.

load_dataset.load_htru1_data(data_dir='htru1-batches-py')