Skip to content

InferenceClient

Bases: ServiceClientWrapper

Wrapper of ServiceClient to handle inference, including: - fitting - prediction

__init__

__init__(service_client=ServiceClient())

fit

fit(X, y) -> str

predict

predict(
    X,
    task: Literal["classification", "regression"],
    train_set_uid: str,
    config=None,
    X_train=None,
    y_train=None,
)