hummingbird.ml.operator_converters.sklearn.sv¶
Converters for scikit-learn SV models: SVC, NuSVC. (LinearSVC is covered by linear_classifier.py).
- hummingbird.ml.operator_converters.sklearn.sv.convert_sklearn_svc_model(operator, device, extra_config)[source]¶
Converter for sklearn.svm.SVC and sklearn.svm.NuSVC
- Args:
operator: An operator wrapping a sklearn.svm.SVC or sklearn.svm.NuSVC model device: String defining the type of device the converted operator should be run on extra_config: Extra configuration used to select the best conversion strategy
- Returns:
A PyTorch model