mednet.models.classifyΒΆ

Modules

alexnet

AlexNet network architecture, from [ALEXNET-2012].

cnn3d

Simple 3D convolutional neural network architecture for classification.

densenet

DenseNet-121 network architecture, from [DENSENET-2017].

logistic_regression

Logistic regression model for multi-class classification.

loss_weights

Helpers for computing (sample/label) weights for loss terms.

mlp

Multi-layer perceptron model for multi-class classification.

model

Definition of base model type for classification tasks.

normalizer

Functions to compute normalisation factors based on dataloaders.

pasa

Simple CNN network model from [PASA-2019].

typing

Definition of types related to models used in classification tasks.