Skip to content
This repository has been archived by the owner on Oct 19, 2024. It is now read-only.

Add models to do classification #106

Merged
merged 16 commits into from
Sep 6, 2024
Merged

Conversation

moerlemans
Copy link
Contributor

@moerlemans moerlemans commented Aug 8, 2024

This PR adds the common models used in digital pathology classification. List of models currently will be

  • attention-based MIL (also known as deepMIL)
  • transMIL
    - wrappers for loading huggingface models/ transforms just by name/path

@moerlemans moerlemans marked this pull request as ready for review September 4, 2024 15:13
@JorenB
Copy link
Contributor

JorenB commented Sep 5, 2024

Should still make TransMIL and ABMIL consistent in terms of initialization parameters

JorenB
JorenB previously approved these changes Sep 5, 2024
Copy link
Contributor

@JorenB JorenB left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, nice and concise

@JorenB JorenB dismissed their stale review September 5, 2024 20:05

several pre-commit checks did not pass

@moerlemans moerlemans merged commit d0ba0e9 into main Sep 6, 2024
1 check passed
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants