RanodmSearchCV in pytorch

Hi, I have a question regarding hyperparameter tunning in PyTorch. I have a neural net in PyTorch that I want to tune its hyperparameters using RandomsearchCv in sklearn. But, I didn’t find any wrapper for pytorch model that makes it possible to use RandomSearchCv in sklearn. Any help is appreacited.


Skorch provides a scikit-learn compatible layer on top of PyTorch.

1 Like