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