Using classifiers in scikit-learn with PyTorch

You could apply these techniques using scikit-learn.
They work an numpy arrays, so if you would like to train an SVM on top of a CNN, you could just get the numpy array from the tensor:

output = pytorch_model(x)
output = output.numpy()
3 Likes