Skip to content

Commit

Permalink
added info to Colab notebook to further explain model training parame…
Browse files Browse the repository at this point in the history
…ters
  • Loading branch information
gkaldev committed Oct 28, 2020
1 parent bfb7491 commit 1baff17
Showing 1 changed file with 13 additions and 1 deletion.
14 changes: 13 additions & 1 deletion Firebase_ML_on_device_recommentations.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -623,7 +623,19 @@
"* Setup callbacks for tensorboard and checkpoint manager.\n",
"* Run model.fit with compiled model, where you could specify number of epochs to train, number of train steps in each epoch and number of eval steps in each epoch.\n",
"\n",
"To start training please execute command. Please note that we are using a very small number of epochs (10) to speed up training time at the expense of model quality. Generating a high quality model often requires a much higher number. For this model, setting num_epochs to at least 100 should provide a model of sufficient quality. \n"
"## Model training parameters\n",
"\n",
"### Encoder type\n",
"\n",
"You can train the model using three different encoder types: a convolutional neural net (cnn), a recurrent neural net (rnn), or a bag of words (bow). You can select between the various types with the **--encoder_type** parameter supplying **cnn**, **rnn**, or **bow**. Different encoders have strengths and weakensses depending on the input / output characteristics of your dataset.\n",
"\n",
"For example: If the input context (here, the user history length) is long, cnn and rnn would be more suitable as they have better summarization ability with longer user histories.\n",
"\n",
"### Training time / size\n",
"\n",
"Another consideration is training time. Rnn generally requires the longer training times, followed by cnn, and finally bow with the shortest training times. Bag of words will also be a smaller sized model if space is a consideration.\n",
"\n",
"To start training, execute the following command. Please note that we are using a very small number of epochs (**num_epochs** parameter below) of 10 to speed up training time at the expense of model quality. Generating a high quality model often requires a much higher number. For this model, setting num_epochs to at least 100 should provide a model of sufficient quality. \n"
]
},
{
Expand Down

0 comments on commit 1baff17

Please sign in to comment.