diff --git a/doc/api.rst b/doc/api.rst index 186f9eeab1aebde1081615f24fabd9971a70777d..f493d80b319435caabdd2e9f409c6bdcf0c2102f 100644 --- a/doc/api.rst +++ b/doc/api.rst @@ -45,6 +45,7 @@ CNN and other models implemented. mednet.models.logistic_regression mednet.models.loss_weights mednet.models.mlp + mednet.models.model mednet.models.normalizer mednet.models.separate mednet.models.transforms