yobx.sklearn.linear_model.ransac#

yobx.sklearn.linear_model.ransac.sklearn_ransac_regressor(g: GraphBuilderExtendedProtocol, sts: Dict, outputs: List[str], estimator: RANSACRegressor, X: str, name: str = 'ransac_regressor') str[source]#

Converts a sklearn.linear_model.RANSACRegressor into ONNX.

After fitting, RANSACRegressor stores the best-fitting sub-estimator in estimator_. Prediction is a direct delegation:

y = estimator_.predict(X)

This converter therefore simply looks up and calls the ONNX converter for the fitted sub-estimator (estimator.estimator_).

Graph structure:

X ──[sub-estimator converter]──► predictions
Parameters:
  • g – the graph builder to add nodes to

  • sts – shapes and types defined by scikit-learn

  • outputs – desired output tensor names (one entry: predictions)

  • estimator – a fitted RANSACRegressor

  • X – name of the input tensor

  • name – prefix used for names of nodes added by this converter

Returns:

name of the predictions output tensor