PyTorchDiscriminator¶
-
class
PyTorchDiscriminator
(n_features=1, n_out=1)[source]¶ Bases:
qiskit.aqua.components.neural_networks.discriminative_network.DiscriminativeNetwork
Discriminator based on PyTorch
- Parameters
n_features (
int
) – Dimension of input data vector.n_out (
int
) – Dimension of the discriminator’s output vector.
- Raises
MissingOptionalLibraryError – Pytorch not installed
Methods
Get data sample labels, i.e. true or fake.
Compute gradient penalty for discriminator optimization
Load discriminator model
Loss function
Save discriminator model
Set seed.
Perform one training step w.r.t to the discriminator’s parameters
Attributes
-
discriminator_net
¶ Get discriminator
- Returns
discriminator object
- Return type
object