Transposition in PyTorch to Sklearn models such as Random Forest or SVM - View it on GitHub