Import all necessary libraries.
Check the available devices (CPU/GPU)
Define a function for embedding sizes. It is discussed in the thesis.
Define a function for plotting purposes.
Read the dataset and prepare its splits. Also calculate the class weight.
Define the hyper-parameters. Can be replaced by grid search or any other mechanism.
Create, design and compile the model.
Train the model.
Save the results and show the plot.