Define a simple PyTorch nn.Module with one hidden layer. Write the training loop including the forward pass, Binary Cross Entropy (BCE) loss calculation, and an Adam optimizer step. Train for 50 epochs on quality_binary and report final test metrics.
Define a simple PyTorch nn.Module with one hidden layer. Write the training loop including the forward pass, Binary Cross Entropy (BCE) loss calculation, and an Adam optimizer step. Train for 50 epochs on quality_binary and report final test metrics.