Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fixed issues in MNIST task by Cecilia Wang #15

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

ccc9991113
Copy link

  1. Model Output Fix: Updated NUM_CLASSES to 10 to correctly predict digits from 0-9.
  2. Hyperparameter Adjustments: Increased EPOCHS to 30 for better training and lowered LEARNING_RATE to 0.001 for smoother optimization.
  3. Early Stopping: Added early stopping to prevent overfitting and save time.
  4. Loss Function: Changed the loss function to sum the batch loss for better control with large batches.
  5. Error Metrics: Added F1 Score, Precision, and Recall for more detailed performance evaluation.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant