# Dataset
## helper function to create data
------------------------------------------------------------------------
source
### sample_spin_configurations
``` python
def sample_spin_configurations(
wave_function, num_samples, N, key
):
```
*samples spin config. from an exact wavefunction*
Args:
wave_function: (2\*\*N,) complex64 num_samples: int, nbr of samples N:
int, nbr of sites key: a JAX PRNGKey
Returns:
spin_configurations: (num_samples, N) int32, samples of spin
configurations
------------------------------------------------------------------------
source
### get_classical_shadow
``` python
def get_classical_shadow(
psi, num_shots, N, rng_key, bases:str='XYZ'
):
```
*generate classical shadows from an exact wavefunction*
Args:
psi: (2\*\*N,) complex64 from, f. ex., exact diagonalization num_shots:
int N: int rng_key: a JAX PRNGKey bases: which measurement basis,
default ‘XYZ’ -\> {X,Y,Z} randomly, other available option: ‘X’, ‘Y’,
‘Z’, ‘XY’, ‘YZ’
Returns:
shadows: (num_shots, 2\*N) int32 interleaved \[basis,outcome\] pairs
basis in {2,3,4} = {X,Y,Z}, outcome in {0,1} = {+1,-1}
## class
------------------------------------------------------------------------
source
### Dataset
``` python
def Dataset(
data:Array, thetas:Sequence, data_type:str, local_dimension:int=0, local_states:Array=None
):
```