moabb icon indicating copy to clipboard operation
moabb copied to clipboard

Advice on extending to PyTorch

Open jinglescode opened this issue 4 years ago • 2 comments

Any suggestions on how to use existing code to work on PyTorch? Because from the examples, evaluation.process depends on sklearn.pipeline, and thus it executes models from sklearn such as sklearn.linear_model import LogisticRegression. Any advice?

jinglescode avatar Dec 29 '20 03:12 jinglescode

I'm interested in this too, looks like one can use something like skorch: https://github.com/skorch-dev/skorch

ErikBjare avatar Dec 29 '20 19:12 ErikBjare

I've never knew about skorch, and it looks good. Let me try training a simple PyTorch network with moabb's evaluation.process.

jinglescode avatar Dec 30 '20 02:12 jinglescode