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

Fix(F1Score): correct computation of F1Score in automl training #226

Merged
merged 1 commit into from
Oct 1, 2024

Conversation

rizoudal
Copy link

@rizoudal rizoudal commented Aug 9, 2024

This pull request addresses issue #224.

Changes made:

  • Added F1Score compution to automl training block.
  • Added labels cast to prevent type errors.

Files modified:

  • aucmedi/automl/block_train.py
  • aucmedi/data_processing/data_generator.py

Please review the changes and let me know if there are any questions or further adjustments needed.

- added F1Score to automl training block
- cast labels to float32 to prevent type errors
@muellerdo
Copy link
Member

F1Score computation of tensorflow is buggy?
Seems like shape is not really passed through from our DataGenerator/Sequence object.

Wait for fix of TF.

Possible hotfix: Setting shape ourselves with add_variable() or something??

@muellerdo muellerdo merged commit bd4cff3 into frankkramer-lab:development Oct 1, 2024
1 of 6 checks passed
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

Successfully merging this pull request may close these issues.

2 participants