Add properties to models
ravih18 opened this issue · 2 comments
ravih18 commented
It would be great to add some properties to the network class such as task (list of tasks associated to the model), is_3D (bool) and adding a description in the docstring of each models available in ClinicaDL.
In that way we could improve the informations displayed in the clinicadl train list_model command line that would help a user (in which case should I use this model, for which task we estimate that a model in well performing, why did we add a model...).
ravih18 commented
An other good information that could be added is the reference paper from which the model has been implemented and in which paper it has been used.