Custom GDS methods

Hey, everyone!

I'm starting using Graph Data Science methods but it looks like this library has solutions for a very small amount of problems

For example, a very simple idea: we have an oriented graph with spreading illness, so that the closer and the stronger our connection to neighbours, the greater the chance for a current node to get ill too

Looks a bit similar to Label propagation, but it's rather a property propagation with an effect of loosing influence with the length of the path

But I have no idea how I can solve it in Neo4j. Is there any way to do this? Is it with the help of GDS?

Thank u so much in advance!