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

Application to multiple classifications #2

Open
Orange2107 opened this issue Nov 3, 2023 · 4 comments
Open

Application to multiple classifications #2

Orange2107 opened this issue Nov 3, 2023 · 4 comments

Comments

@Orange2107
Copy link

Hi @njmarko
Have you tried applying the model to a multiclassification of 964 side effects?

@njmarko
Copy link
Owner

njmarko commented Nov 3, 2023

Hi @Orange2107

I haven't tried that yet because I currently don't have the hardware to perform such experiments.

It is possible to do something like that, and the authors of the Drug-Drug Adverse Effect Prediction with Graph Co-Attention paper tried it with their model on the same dataset.

I tried my model on the same dataset with their dataloader, and with some modifications, I believe it could be tweaked to work for multiclassification of side effects. You can check out how the authors of that paper managed to do it for their model in their code.

@Orange2107
Copy link
Author

Hi @njmarko
I'm glad you were able to reply.
I couldn't find a section on multi- classification after looking up her code.
Thank you

@njmarko
Copy link
Owner

njmarko commented Nov 3, 2023

I did a quick search in that repo, and I also couldn't find it.

I only found cross entropy in the embedded repo.

It is mentioned in the paper though.

image
image

@Orange2107
Copy link
Author

@njmarko
Yeah, I followed this and turned the output of your model into 964 dimensions. I removed the neg set and used the pos set for multiclassification. I get high AUROC but very low AUPRC.
I'm trying to figure that out now.

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