|
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950 |
- # 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.
-
|