mednet.models.classify

Specialized classification models and associated functionality.

Modules

alexnet

AlexNet network architecture, from [KSH17].

cnn3d

Simple 3D convolutional neural network architecture for classification.

densenet

DenseNet-121 network architecture, from [HLVDMW17].

logistic_regression

Logistic regression model for multi-class classification.

mlp

Multi-layer perceptron model for multi-class classification.

model

Definition of base model type for classification tasks.

pasa

Simple CNN network model from [PGP+19].

typing

Definition of types related to models used in classification tasks.