Add the possibility to put more weight to EOS (except classif)

[pep8speaks]

Hello @EmmaRenauld! Thanks for opening this PR.

• In the file dwi_ml/models/direction_getter_models.py:

Line 203:80: E501 line too long (80 > 79 characters)

• In the file dwi_ml/models/utils/direction_getters.py:

Line 68:80: E501 line too long (80 > 79 characters)

Do see the PEP 8 -- Style Guide for Python Code