Provides a torch::nn_module with a simple fully connected neural net, for use as the default architecture for tabular data in RGAN.
Discriminator(
data_dim,
hidden_units = list(128, 128),
dropout_rate = 0.5,
sigmoid = FALSE
)
The number of columns in the data set
A list of the number of neurons per layer, the length of the list determines the number of hidden layers
The dropout rate for each hidden layer
Switch between a sigmoid and linear output layer (the sigmoid is needed for the original GAN value function)
A torch::nn_module for the Discriminator