내가 보려고 만든 블로그

<Recommendation> MF , Biased_MF 구현 본문

Recommendation/모델링

<Recommendation> MF , Biased_MF 구현

정의김 2022. 7. 23. 20:47

MF 랑 Biased_MF 구현하였다.

1. 토치로 구현하였고 옵티마이저는 Adam 사용, ALS는 나중에

2. 특별한건없구 Biased_MF의 경우 구현할때 biased 텐서를 유저, 아이템 각각에 대해 생성해주고

만들어진 Reconstruction 에 더해주는 방법으로 쉽게 구현할수 있다.

 

코드는 요기

https://github.com/todtjs92/Search_and_Recommendation/tree/master/MF