Skip to content

Effect on performance from class imbalance in a multiclass-classification setting? #19

@shashankg7

Description

@shashankg7

Hi,

I am trying out this model on my custom dataset with the following frequency distribution of class labels :

7: 23849, 0: 15159, 1: 6445, 4: 5759, 5: 3969, 3: 3659, 2: 2845, 6: 492

I am getting ~65% accuracy on ~16K testing samples after training on the above mentioned dataset. Can class imbalance be one of the reason for this low accuracy?

I am using the model in it's original setting (assuming the best settings as reported in the paper).

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions