如何让grid_search上的快速学习者？

Question

how to grid_search to deep learning fastai (我使用LSTM)来调优Learner - https://docs.fast.ai/basic_train.html#Learner的超参数我的代码是：

from fastai.train import Learner
from fastai.train import DataBunch

model = NeuralNet(embedding_matrix, y_aux_train.shape[-1])
learn = Learner(databunch, model, loss_func=custom_loss)   

我想知道如何使用-https://scikit-learn.org/stable/modules/grid_search.html调优超参数，就像在sklearn的标准分类算法中一样。

Answer 1

您可以使用https://github.com/skorch-dev/skorch，它允许您在pytorch模型上使用Sklearn网格搜索，然后将此模型包装在快速学习器中。