notice
This is documentation for Rasa Open Source Documentation v2.1.x, which is no longer actively maintained.
For up-to-date documentation, see the latest version (2.5.x).
rasa.utils.tensorflow.models
RasaModel Objects
Completely override all public methods of keras Model.
Cannot be used as tf.keras.Model
__init__
Initialize the RasaModel.
Arguments:
random_seed
- set the random seed to get reproducible results
fit
Fit model data
train_on_batch
Train on batch
batch_to_model_data_format
Convert input batch tensors into batch data format.
Batch contains any number of batch data. The order is equal to the key-value pairs in session data. As sparse data were converted into indices, data, shape before, this methods converts them into sparse tensors. Dense data is kept.
linearly_increasing_batch_size
Linearly increase batch size with every epoch.
The idea comes from https://arxiv.org/abs/1711.00489.