def test_qgan_save_model(self):
"""Test the QGAN functionality to store the current model."""
# Set number of qubits per data dimension as list of k qubit values[#q_0,...,#q_k-1]
num_qubits = [2]
# Batch size
batch_size = 100
# Set number of training epochs
num_epochs = 5
with tempfile.TemporaryDirectory() as tmpdirname:
_qgan = QGAN(
self._real_data,
self._bounds,
num_qubits,
batch_size,
num_epochs,
discriminator=NumPyDiscriminator(n_features=len(num_qubits)),
snapshot_dir=tmpdirname)
_qgan.seed = self.seed
_qgan.set_generator()
trained_statevector = _qgan.run(
QuantumInstance(BasicAer.get_backend('statevector_simulator'),
seed_simulator=algorithm_globals.random_seed,
seed_transpiler=algorithm_globals.random_seed))
trained_qasm = _qgan.run(
QuantumInstance(BasicAer.get_backend('qasm_simulator'),
seed_simulator=algorithm_globals.random_seed,
seed_transpiler=algorithm_globals.random_seed))
self.assertAlmostEqual(trained_qasm['rel_entr'],
trained_statevector['rel_entr'],
delta=0.1)
def test_qgan_training_run_algo_numpy(self):
"""Test QGAN training using a NumPy discriminator."""
# Set number of qubits per data dimension as list of k qubit values[#q_0,...,#q_k-1]
num_qubits = [2]
# Batch size
batch_size = 100
# Set number of training epochs
num_epochs = 5
_qgan = QGAN(
self._real_data,
self._bounds,
num_qubits,
batch_size,
num_epochs,
discriminator=NumPyDiscriminator(n_features=len(num_qubits)),
snapshot_dir=None)
_qgan.seed = self.seed
_qgan.set_generator()
trained_statevector = _qgan.run(
QuantumInstance(BasicAer.get_backend('statevector_simulator'),
seed_simulator=algorithm_globals.random_seed,
seed_transpiler=algorithm_globals.random_seed))
trained_qasm = _qgan.run(
QuantumInstance(BasicAer.get_backend('qasm_simulator'),
seed_simulator=algorithm_globals.random_seed,
seed_transpiler=algorithm_globals.random_seed))
self.assertAlmostEqual(trained_qasm['rel_entr'],
trained_statevector['rel_entr'],
delta=0.1)
def test_qgan_training_run_algo_numpy_multivariate(self):
"""Test QGAN training using a NumPy discriminator, for multivariate distributions."""
# Set number of qubits per data dimension as list of k qubit values[#q_0,...,#q_k-1]
num_qubits = [1, 2]
# Batch size
batch_size = 100
# Set number of training epochs
num_epochs = 5
# Reshape data in a multi-variate fashion
# (two independent identically distributed variables,
# each represented by half of the generated samples)
real_data = self._real_data.reshape((-1, 2))
bounds = [self._bounds, self._bounds]
_qgan = QGAN(
real_data,
bounds,
num_qubits,
batch_size,
num_epochs,
discriminator=NumPyDiscriminator(n_features=len(num_qubits)),
snapshot_dir=None,
)
_qgan.seed = self.seed
_qgan.set_generator()
trained_statevector = _qgan.run(
QuantumInstance(
BasicAer.get_backend("statevector_simulator"),
seed_simulator=algorithm_globals.random_seed,
seed_transpiler=algorithm_globals.random_seed,
))
trained_qasm = _qgan.run(
QuantumInstance(
BasicAer.get_backend("qasm_simulator"),
seed_simulator=algorithm_globals.random_seed,
seed_transpiler=algorithm_globals.random_seed,
))
self.assertAlmostEqual(trained_qasm["rel_entr"],
trained_statevector["rel_entr"],
delta=0.1)
def test_qgan_gradient_penalty_pytorch(self):
"""
Test QGAN training with gradient penalty for the discriminator
"""
num_qubits = [2]
# Batch size
batch_size = 100
# Set number of training epochs
num_epochs = 5
_qgan = QGAN(
self._real_data,
self._bounds,
num_qubits,
batch_size,
num_epochs,
discriminator=PyTorchDiscriminator(n_features=len(num_qubits)),
snapshot_dir=None,
penalty=True,
)
_qgan.seed = self.seed
_qgan.set_generator()
trained_statevector = _qgan.run(
QuantumInstance(
BasicAer.get_backend("statevector_simulator"),
seed_simulator=algorithm_globals.random_seed,
seed_transpiler=algorithm_globals.random_seed,
))
trained_qasm = _qgan.run(
QuantumInstance(
BasicAer.get_backend("qasm_simulator"),
seed_simulator=algorithm_globals.random_seed,
seed_transpiler=algorithm_globals.random_seed,
))
self.assertAlmostEqual(trained_qasm["rel_entr"],
trained_statevector["rel_entr"],
delta=0.1)
