Is this limited to just sklearn models? #33
Description
Would like to get it working with models that arent always constrained to X,y fit and predict like sklearn models. Such models include Prophet, ARIMA from statsmodels, and pytorch models in their different flavours. Are there any suggested work around for this? The rigid X y specification really limits use of this with all types of models