User improvement and patch classification problem

UX: Improving Warning Messages for Improper Use

I have focused on enhancing the user experience by providing more helpful warning messages when the tool is used improperly.

Classification Boolean Variable to Correctly Assess Neural Network Designs

I introduced the classification variable (defaulted to True) in the constructor of the PytorchWrapper class. Calculating the overall accuracy of a model only makes sense in classification problems and not in regression analysis. This update ensures a more accurate assessment of the type of neural network being designed.

Additional Changes

I have cleaned up unnecessary comments and improved the descriptions of functions and classses for better clarity and maintainability.