Text Classification

Text Classification - Jupyter Notebook

The example text_classification_AG_NEWS.ipynb demonstrates using Juypter Notebook for Trains, and the integration of Trains into code which trains a network to classify text in the torchtext AG_NEWS dataset, and then applies the model to predict the classification of sample text. Trains automatically logs the scalar and console output by calling TensorBoard methods. In the example, we explicitly log parameters with the Task. When the script runs, it creates an experiment named text classifier which is associated with the Text Example project.


Accuracy, learning rate, and training loss appear in the RESULTS tab, SCALARS tab, along with the resource utilization plots, which are titled :monitor: machine.


Explicitly connect parameters to the Task by creating a parameter dictionary and calling Task.connect providing that dictionary.

configuration_dict = {'number_of_epochs': 6, 'batch_size': 16, 'ngrams': 2, 'base_lr': 1.0}
configuration_dict = task.connect(configuration_dict)  # enabling configuration override by trains

Since the example script uses TensorFlow, TensorFlow DEFINES are also logged.


Text printed to the console for training progress, as well as all other console output, appear in the RESULTS tab, LOG tab.