Mentions légales du service

Skip to content
Snippets Groups Projects
Verified Commit 02afef6e authored by ANDREY Paul's avatar ANDREY Paul
Browse files

Improve dtype handling in 'SklearnSGDModel'.

- Scikit-Learn 1.3.0 introduced the possibility to use any dtype for
  SGD-based models' weights.
- As a consequence, this commit introduces the optional 'dtype' argument
  to 'SklearnSGDModel.__init__' and '.from_parameters' methods, which is
  only used with `sickit-learn >=1.3` and an un-initialized model.
- Coherently (and for all scikit-learn versions), additional type/dtype
  verification has been implemented under the hood of `_unpack_batch`.
parent 3b687a3a
No related branches found
No related tags found
1 merge request!50Fix compatibility issues with scikit-learn 1.3.0 and tensorflow 2.13.0
Pipeline #830752 failed
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment