My work includes movie recommendations using collaborative filtering which is the most accurate way to recommend things. I have experimented with the size of embedding layer , batch_size and learning rate much and came to the final conclusion that particularly for this dataset the embedding layer of 30 was best in the sizes of 30,50 and 64 . Also the mean square error was better when kept layer size small. This might be due to the fact that the data might not be that complex to have feature extraction of upto 50 or 64. Extracting features upto this limit might have led the model to memorize and thus overfitting the data. The best mean square error achieves was of 0.8254.
-
Notifications
You must be signed in to change notification settings - Fork 0
My work includes movie recommendations using collaborative filtering with a least mean square error of 0.8254. Also Experimented on embedded matrices of different sizes and feature extraction. I have experimented with the size of embedding layer , batch_size and learning rate much and came to the final conclusion that particularly for this datas…
deadskull7/Movie-Recommendations-using-Collaborative-Filtering
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
My work includes movie recommendations using collaborative filtering with a least mean square error of 0.8254. Also Experimented on embedded matrices of different sizes and feature extraction. I have experimented with the size of embedding layer , batch_size and learning rate much and came to the final conclusion that particularly for this datas…
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published