I already written the code but I want it to be rewritten to be more accurate and

I already written the code but I want it to be rewritten to be more accurate and clear with no ChatGPT dedicated if found. I attached the files need mine is COE292 and the original assignment is named assignment4
• The assignment is about the Convolution Neural network
• Use the attached notebook file to solve the assignment (check the first reply)
• Submit your solution as a python file (.py) not as a notebook (.ipynb). Your submission will not be graded if you submit a notebook (i.e. you will get a 0 if you submit a .ipynb file). Follow the submission direction in the attached notebook to know how to convert a .ipynb file to .py file