How to wrap a deep learning model

[inoueakimitsu]

start_date: 2021/10/16
due_date: 2021/10/16
progress: 0
parent: 0
dependon:
  - '1'

We need to create a procedure to wrap deep learning models.

With an interface similar to scikit-learn, there is no need to change the current implementation.

For Keras, you can use keras.wrappers.scikit_learn.KerasClassifier.

For PyTorch, I do not know how to do this at the moment.

[inoueakimitsu]

You can use skorch for PyTorch.

We need to investigate lightning as well.

[inoueakimitsu]

I should write a sample code using skorch and milwrap.