Movie recommendation is based on the neural collaborative filtering technique. Movielens dataset consisting of 100K records is analyzed.
- Data Analysis and manipulation using pandas
- Data Visualization using matplotlib
- Test Train split
- Created movie embedding path
- Created user embedding path
- Concatenated movie and user vectors
- Weight initialization using the random uniform technique
- Added 5 fully connected layers with activation function as ReLu
- Adam optimizer used
- Model has complied
- Model is run for 18 epochs with verbosity 1
- Achieved loss of 0.6
- As seen below graph of Training error vs epoch, there is a reduction in error continuously after epoch=5
- Minimum RSE of 0.8 is at epoch 18
Email - [email protected]
LinkedIn - https://www.linkedin.com/in/tejas-ta/
Blogs - https://tejasta.medium.com/