diff --git a/docs/readthedocs/requirements-doc.txt b/docs/readthedocs/requirements-doc.txt index 167b9b464fe..7e860383a38 100644 --- a/docs/readthedocs/requirements-doc.txt +++ b/docs/readthedocs/requirements-doc.txt @@ -20,11 +20,13 @@ commonmark==0.8.1 recommonmark==0.5.0 readthedocs-sphinx-ext<2.2 sphinx_rtd_theme==0.5.2 -scikit-learn==0.22.2.post1 +scikit-learn==1.0.2 pystan==2.19.1.1 prophet pmdarima sphinx_markdown_tables numpy==1.21.2 xgboost -torchmetrics \ No newline at end of file +torchmetrics +pickle5==0.0.12 +protobuf~=3.19.0 \ No newline at end of file diff --git a/docs/readthedocs/source/doc/PythonAPI/Chronos/automodels.rst b/docs/readthedocs/source/doc/PythonAPI/Chronos/automodels.rst index b650121f7ef..faff1a74bfc 100644 --- a/docs/readthedocs/source/doc/PythonAPI/Chronos/automodels.rst +++ b/docs/readthedocs/source/doc/PythonAPI/Chronos/automodels.rst @@ -5,7 +5,7 @@ AutoTCN ------------------------------------------- AutoTCN is a TCN forecasting model with Auto tuning. -Other API follows its base class(BasePytorchAutomodel). +Other API follows its base class(BaseAutomodel). .. automodule:: bigdl.chronos.autots.model.auto_tcn :members: @@ -17,7 +17,7 @@ AutoLSTM ---------------------------------------- AutoLSTM is an LSTM forecasting model with Auto tuning. -Other API follows its base class(BasePytorchAutomodel). +Other API follows its base class(BaseAutomodel). .. automodule:: bigdl.chronos.autots.model.auto_lstm :members: @@ -28,7 +28,7 @@ AutoSeq2Seq ---------------------------------------- AutoSeq2Seq is an Seq2Seq forecasting model with Auto tuning. -Other API follows its base class(BasePytorchAutomodel). +Other API follows its base class(BaseAutomodel). .. automodule:: bigdl.chronos.autots.model.auto_seq2seq :members: @@ -55,10 +55,10 @@ AutoProphet is a Prophet forecasting model with Auto tuning. :undoc-members: :show-inheritance: -BasePytorchAutomodel +BaseAutomodel ------------------------------------------------------------ AutoLSTM, AutoSeq2Seq and AutoTCN all follow the same API as stated below. -.. autoclass:: bigdl.chronos.autots.model.base_automodel.BasePytorchAutomodel +.. autoclass:: bigdl.chronos.autots.model.base_automodel.BaseAutomodel :members: :show-inheritance: diff --git a/python/chronos/src/bigdl/chronos/autots/tspipeline.py b/python/chronos/src/bigdl/chronos/autots/tspipeline.py index 175dc1a2c1e..9b34451f9f7 100644 --- a/python/chronos/src/bigdl/chronos/autots/tspipeline.py +++ b/python/chronos/src/bigdl/chronos/autots/tspipeline.py @@ -21,7 +21,6 @@ from bigdl.chronos.data import TSDataset from bigdl.chronos.metric.forecast_metrics import Evaluator -from bigdl.chronos.pytorch.utils import _pytorch_fashion_inference DEFAULT_MODEL_INIT_DIR = "model_init.ckpt" DEFAULT_BEST_MODEL_DIR = "best_model.ckpt"