Skip to content

fix: default to 4 num_classes - #36

Merged
GabrielLidenor merged 1 commit into
mainfrom
35-fixconfig-update-default-num_classes-from-7-to-4-for-4-class-emotion-dataset
Aug 30, 2026
Merged

fix: default to 4 num_classes#36
GabrielLidenor merged 1 commit into
mainfrom
35-fixconfig-update-default-num_classes-from-7-to-4-for-4-class-emotion-dataset

Conversation

@GabrielLidenor

@GabrielLidenor GabrielLidenor commented Aug 30, 2026

Copy link
Copy Markdown
Owner

📝 Description / Why


The default num_class was 7 for the model, but our dataset only has 4.

🛠️ PR Changes


  • src/models/emote_vision_model.py - update default num_classes to 4

🧪 Verification & Testing Completed


📌 Reviewer Checklist

  • Code follows the modular structural design of the project.
  • Dependencies are cleanly synced and added to requirements.txt if new packages were imported.
  • No local datasets, binary files, or credentials are being tracked (Double check .gitignore).
  • Pipeline executes end-to-end without throwing runtime exceptions.

Because our dataset only has 4 types of feelings
@GabrielLidenor GabrielLidenor self-assigned this Aug 30, 2026
@GabrielLidenor
GabrielLidenor merged commit 8296767 into main Aug 30, 2026
1 check passed
@GabrielLidenor
GabrielLidenor deleted the 35-fixconfig-update-default-num_classes-from-7-to-4-for-4-class-emotion-dataset branch August 30, 2026 22:43
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.

fix(config): Update default num_classes from 7 to 4 for 4-class emotion dataset

1 participant