如何在fastai学习者上进行grid_search？

Question

如何使grid_search成为深度学习Fastai（我使用LSTM）来调整学习者的超级性能-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网格搜索，然后将此模型包装在fastai学习器中。