haji_thesis/Tools

Cell 1:

Lines 1~13:

Import all necessary libraries.

Line 14:

Check the available devices (CPU/GPU)

Cell 2:

Lines 1~2:

Define a function for embedding sizes. It is discussed in the thesis.

Lines 4~54:

Define a function for plotting purposes.

Cell 3:

Lines 1~9:

Read the dataset and prepare its splits. Also calculate the class weight.

Cell 4:

Lines 1~12:

Define the hyper-parameters. Can be replaced by grid search or any other mechanism.

Lines 13~69:

Create, design and compile the model.

Lines 70~80:

Train the model.

Lines 81~97:

Save the results and show the plot.