v3.5.0
neuml/txtaiv3.5.0Oct 18, 2021by davidmezzetti
AI Summary
Focuses on ONNX export capabilities, adds registry methods for auto models, and integrates scikit-learn support for model export.
Key Highlights
- ONNX export pipeline added
- Registry methods for auto models
- Scikit-learn to ONNX export support
New Features
- Scikit-learn to ONNX export pipeline
- Registry methods for auto models
- Notebook for loading scikit-learn and PyTorch models
- Parameter to return raw model outputs for labels pipeline
- Parameter to use standard pooling for TransformersVectors
Full Release Notes
This release adds the following new features, improvements and bug fixes. New Features -------------------------- - Add scikit-learn to ONNX export pipeline (#124) - Add registry methods for auto models (#126) - Add notebook to demonstrate loading scikit-learn and PyTorch models (#127) Improvements -------------------------- - Add parameter to return raw model outputs for labels pipeline (#123) - Add parameter to use standard pooling for TransformersVectors (#125) Bug Fixes -------------------------- - Pass model configuration to ONNX Models (#121) - Fix incorrect import in Notebooks (#122)