hummingbird.ml.operator_converters.sklearn.mlp

Converters for scikit-learn MLP models: MLPClassifier, MLPRegressor

hummingbird.ml.operator_converters.sklearn.mlp.convert_sklearn_mlp_classifier(operator, device, extra_config)

Converter for sklearn.neural_network.MLPClassifier

Args:

operator: An operator wrapping a sklearn.neural_network.MLPClassifier 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

hummingbird.ml.operator_converters.sklearn.mlp.convert_sklearn_mlp_regressor(operator, device, extra_config)

Converter for sklearn.neural_network.MLPRegressor

Args:

operator: An operator wrapping a sklearn.neural_network.MLPRegressor 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