Data Handling

qucumber.utils.data.extract_refbasis_samples(train_samples, train_bases)[source]

Extract the reference basis samples from the data.

Parameters:
  • train_samples (numpy.array) – The training samples.
  • train_bases (numpy.array) – The bases of the training samples.
Returns:

The samples in the data that are only in the reference basis.

Return type:

torch.Tensor

qucumber.utils.data.load_data(tr_samples_path, tr_psi_path=None, tr_bases_path=None, bases_path=None)[source]

Load the data required for training.

Parameters:
  • tr_samples_path (str) – The path to the training data.
  • tr_psi_path (str) – The path to the target/true wavefunction.
  • tr_bases_path (str) – The path to the basis data.
  • bases_path (str) – The path to a file containing all possible bases used in the tr_bases_path file.
Returns:

A list of all input parameters.

Return type:

list