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.