Implement SKlearn API compatibility testing for ml.FullyConnectedEstimator

applied-ml-bde / bde

Bayesian Deep Ensembles

https://applied-ml-bde.github.io/bde/

BSD 3-Clause "New" or "Revised" License

2 stars 0 forks source link

Implement SKlearn API compatibility testing for ml.FullyConnectedEstimator #20

Closed cklyne closed 2 months ago

cklyne commented 2 months ago

SKlearn includes automatic tests for API compatibility for custom estimators. This test be implemented and we should make sure that ml.FullyConnectedEstimator passes it.

There might be some issues when using this test with jax and jit.