diff --git a/nobrainer/models/__init__.py b/nobrainer/models/__init__.py index 8dd75b44..a4842bc7 100644 --- a/nobrainer/models/__init__.py +++ b/nobrainer/models/__init__.py @@ -55,6 +55,6 @@ def get(name): def available_models(): return list(_models) + def list_available_models(): pprint(available_models()) - diff --git a/nobrainer/processing/segmentation.py b/nobrainer/processing/segmentation.py index ff0519df..7ba9a32a 100644 --- a/nobrainer/processing/segmentation.py +++ b/nobrainer/processing/segmentation.py @@ -133,6 +133,7 @@ def predict(self, x, batch_size=1, normalizer=None): batch_size=batch_size, normalizer=normalizer, ) + @classmethod def list_available_models(cls): list_available_models()