Fix prediction serialization issues

Yard1 / ray-skorch

Distributed skorch on Ray Train

Apache License 2.0

57 stars 3 forks source link

Closed Yard1 closed 2 years ago

Yard1 commented 2 years ago

Fixes a serialization error when running predict_proba, makes sure the examples in readme are tested in CI.