secml.ml.model_zoo

load_model

secml.ml.model_zoo.load_model.load_model(model_id)[source]

Load a pre-trained classifier.

Returns a pre-trained SecML classifier given the id of the model.

The following models are available:
  • mnist-svm,

    multiclass CClassifierSVM trained on MNIST

  • mnist59-svm,

    multiclass CClassifierSVM with RBF Kernel trained on MNIST59

  • mnist59-svm-rbf,

    multiclass CClassifierSVM with RBF Kernel trained on MNIST59

  • mnist159-cnn,

    CClassifierPyTorch CNN trained on MNIST159

Parameters
model_idstr

Identifier of the pre-trained model to load.

Returns
CClassifier

Desired pre-trained model.