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

HOW TO USE THIS ANALYSIS FOR MORE THEN 2 CLASSES #3

Open
mansabshebi opened this issue Dec 13, 2020 · 2 comments
Open

HOW TO USE THIS ANALYSIS FOR MORE THEN 2 CLASSES #3

mansabshebi opened this issue Dec 13, 2020 · 2 comments

Comments

@mansabshebi
Copy link

For example, if we have 3 classes to classify
CLASS1=POSITIVE
CLASS2=NEUTRAL
CLASS3=NEGATIVE.
what should be the major changes for multi-class classification using transformers? this query is not an issue this Is my request for more understanding about your work sir. BEST

@YJiangcm
Copy link
Owner

Firstly, You need to do modify in "model.py": change num_labels to 3.

If the number of classes is more than 2, then AUC can not be computered by the original code. So you may also do some other modify according to this.

@mansabshebi
Copy link
Author

sir if you are available could you please provide your email address ?

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

No branches or pull requests

2 participants