ONNX ScikitLearn

flytekitplugins-onnxscikitlearn

FlytekitModel Servingonnxscikit-learnsklearn

This plugin allows you to generate ONNX models from your ScikitLearn models.

Install

pip install flytekitplugins-onnxscikitlearn

Quick Start(example, may need adjustment)

pip install flytekitplugins-onnxscikitlearn

from flytekit import task, workflow
from flytekitplugins.onnxscikitlearn import ScikitLearn2ONNX, ScikitLearn2ONNXConfig

config = ScikitLearn2ONNX(...)

@task
def my_task() -> None:
    ...

@workflow
def my_workflow() -> None:
    my_task()

Available Imports (2)

configScikitLearn2ONNX

ScikitLearn2ONNXConfig is the config used during the scikitlearn to ONNX conversion.

extends dataclass — configuration or data structure for plugin setup

from flytekitplugins.onnxscikitlearn import ScikitLearn2ONNX

configScikitLearn2ONNXConfig

ScikitLearn2ONNXConfig is the config used during the scikitlearn to ONNX conversion.

extends dataclass — configuration or data structure for plugin setup

from flytekitplugins.onnxscikitlearn import ScikitLearn2ONNXConfig

Dependencies

skl2onnxnetworkxonnx

Related Plugins

ONNX PyTorch

Model Serving

This plugin allows you to generate ONNX models from your PyTorch models.

ONNX TensorFlow

Model Serving

This plugin allows you to generate ONNX models from your TensorFlow Keras models.

Dgxc-lepton

Model Serving

A professional Flytekit plugin that enables seamless deployment and management of AI inference endpoints using Lepton AI infrastructure within Flyte workflows.