Open inoueakimitsu opened 2 years ago
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.
You can use skorch for PyTorch.
We need to investigate lightning as well.
I should write a sample code using skorch and milwrap.
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.