feat: pass batch size to predict in dynamic deephit

[HLasse]

Description

This PR adds the option to supply a batch_size argument to the predict method of dynamic deephit that will be passed down to predict_survival. This can drastically speed up evaluation.

Affected Dependencies

NA

How has this been tested?

pytest

Checklist

• I have followed the Contribution Guidelines and Code of Conduct
• I have commented my code following the van der Schaar Lab Styleguide
• I have labelled this PR with the relevant Type labels Don't have the rights to do this
• My changes are covered by tests

[bcebere]

Thank you for your contribution!
At the same, I recommend watching this repo https://github.com/vanderschaarlab/temporai
It is dedicated to prediction models, and in the following days it will get a much stable DynamicDeephit variant.

[HLasse]

Fantastic, looking forward to it! Evgeny directed me here for the most up-to-date version of DynamicDeephit while temporai was getting ready. Please let me know when temporai is ready for use, excited to give it a test!

[DrShushen]

@HLasse a version of temporai with Dynamic DeepHit implementation will be coming end of this week.