hummingbird.ml.operator_converters.prophet

Converter for Prophet.

class hummingbird.ml.operator_converters.prophet.Prophet(*args: Any, **kwargs: Any)

Bases: PhysicalOperator, Module

Class implementing Prophet operator in PyTorch.

_abc_impl = <_abc_data object>

forward(x)

hummingbird.ml.operator_converters.prophet.convert_prophet(operator, device=None, extra_config={})

Converter for prophet.Prophet

Args:

operator: An operator wrapping a prophet.Prophet 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