mlflow / recipes-regression-template

Template repo for kickstarting recipes for regression use case

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Is this limited to just sklearn models?

RussellSB opened this issue · comments

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