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

Enhancement: work for 2 classes #5

Closed
akshayuppal3 opened this issue May 15, 2019 · 1 comment
Closed

Enhancement: work for 2 classes #5

akshayuppal3 opened this issue May 15, 2019 · 1 comment

Comments

@akshayuppal3
Copy link

akshayuppal3 commented May 15, 2019

can this be expanded for 2 classes instead of 5 sentiment classes? Would it require to change LSTM_bidi file?

@ArrasL
Copy link
Owner

ArrasL commented Jun 12, 2019

Yes you can use the present code with 2 or more classes.
For that purpose you don't even need to change the code (the number of classes will be inferred from the size of the output layer weights).

@ArrasL ArrasL closed this as completed Jun 12, 2019
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