tlt.models.text\_classification.text\_classification\_model.TextClassificationModel
===================================================================================

.. currentmodule:: tlt.models.text_classification.text_classification_model

.. autoclass:: TextClassificationModel

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~TextClassificationModel.__init__
      ~TextClassificationModel.benchmark
      ~TextClassificationModel.evaluate
      ~TextClassificationModel.export
      ~TextClassificationModel.load_from_directory
      ~TextClassificationModel.optimize_graph
      ~TextClassificationModel.predict
      ~TextClassificationModel.quantize
      ~TextClassificationModel.train
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~TextClassificationModel.dropout_layer_rate
      ~TextClassificationModel.framework
      ~TextClassificationModel.learning_rate
      ~TextClassificationModel.model_name
      ~TextClassificationModel.num_classes
      ~TextClassificationModel.preprocessor
      ~TextClassificationModel.use_case