Add CNN class for MNIST handling and import it to main.py

[sweep-nightly]

Description

This PR adds a new CNN class in the src/cnn.py file to handle the MNIST dataset. The CNN class is implemented using PyTorch and includes methods for defining the architecture, training the model, and making predictions. The main.py file is modified to import the CNN class and use it to train a model on the MNIST dataset.

Summary of Changes

• Created a new file src/cnn.py to contain the CNN class.
• Imported necessary PyTorch modules in src/cnn.py.
• Defined the CNN class with an appropriate architecture for the MNIST dataset.
• Implemented methods for initializing the CNN, performing the forward pass, training the model, and making predictions.
• Added docstrings and comments to the code, following the user's preferences.
• Modified src/main.py to import the CNN class.
• Instantiated the CNN class and used it to train a model on the MNIST dataset.
• Updated docstrings and comments in src/main.py to reflect the changes made.

Please review and merge this PR.

Fixes #9.

---

🎉 Latest improvements to Sweep:

• Sweep can now passively improve your repository! Check out Rules to learn more.

---

💡 To get Sweep to edit this pull request, you can:

• Comment below, and Sweep can edit the entire PR
• Comment on a file, Sweep will only modify the commented file
• Edit the original issue to get Sweep to recreate the PR from scratch

[sweep-nightly]

Rollback Files For Sweep

• Rollback changes to src/main.py
• Rollback changes to src/cnn.py

[sweep-nightly]

Apply Sweep Rules to your PR?

• Apply: All docstrings and comments should be up to date.
• Apply: Code should be properly formatted and indented.
• Apply: Variable and function names should be descriptive and follow a consistent naming convention.
• Apply: Imports should be organized and grouped together.
• Apply: There should be no unused imports or variables.
• Apply: Code should be properly commented and include docstrings for functions and classes.