.. Template is based on github.com/jdb78/pytorch-forecasting/tree/v0.9.2/docs/source under MIT License

etna.models.nn.TFTModel
=======================

.. currentmodule:: etna.models.nn

.. autoclass:: TFTModel

   
   

   
   
   
   

   

   
   .. rubric:: Methods
   .. autosummary::
   
      ~TFTModel.fit
   
      ~TFTModel.forecast
   
      ~TFTModel.get_model
   
      ~TFTModel.load
   
      ~TFTModel.params_to_tune
   
      ~TFTModel.predict
   
      ~TFTModel.save
   
      ~TFTModel.set_params
   
      ~TFTModel.to_dict
   
   

   

   
   

   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   

   
   .. rubric:: Attributes

   
   This class stores its :code:`__init__` parameters as attributes.
   

   .. autosummary::
   
      ~TFTModel.context_size
   
      ~TFTModel.trainer_params
   
      ~TFTModel.dataset_builder
   
      ~TFTModel.train_batch_size
   
      ~TFTModel.test_batch_size
   
      ~TFTModel.encoder_length
   
      ~TFTModel.trainer