🍀 Challenge: Building the Discriminator

Problem statement

A partial Discriminator class is provided to you. Your task is to define the:

• Network architecture

  • The input layer should have 5 nodes.
  • The hidden layer should have 3 nodes with the Sigmoid activation applied to the output.
  • The output layer should have 1 node with the Sigmoid activation function applied to the output.

• Loss function as MSE.

• Optimiser as SGD. The learning rate should be 0.01.