How can I use KNN, Random Forest models in Pytorch?


(P Ydv) #1

This may seems like an X Y problem, but initially I had huge data and I was not able to train in given resources (RAM problem). So I thought I could use batch feature of Pytorch. But I want to use Methods like KNN, Random Forest, Clustering except Deep Learning. So is it possible or can I use scikit libraries?


(Thomas V) #2

Skorch aims at providing sklearn functions in a PyTorch basis.
That said, if there is something you need that it does not provide, sklearn is a great library and converting Tensors to NumPy arrays is seamless as long as you don’t need gradients flowing through the converted parts.

Best regards

Thomas


(P Ydv) #3

But I think Skorch, does not provide RNN, Random Forest. It only helps in Deep Neural Network.