Skip to content

Collaborative Denoising Auto Encoder with PyTorch

Notifications You must be signed in to change notification settings

bbangjooo/CDAE-PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Main Files

  • Data.py - Top-k 추천 모델을 구현한 적은 처음이라 데이터를 가공하는데에 더 신경을 썼다. Explicit한 rating 정보를 Implicit feedback을 표현하는 바이너리 형태로 바꾸고 Dataframe을 sparse matrix로 바꾸는 작업이 주를 이룬다.

  • models/CDAE.py - CDAE 모델 클래스가 있는 파일이다. 추후에 다른 모델을 만들 때도 BaseModel 구조를 만들고 함께 관리하여 패키지화 시키려고 한다.

  • utils/Evaluator.py - 평가를 위한 클래스가 있는 파일이다. Top-k 추천 모델의 성능을 평가하는 여러 지표가 있는데 그 중 precision과 recall을 사용하였다.

TODO

평가 결과가 예상보다 좋지 않았다. 개인적인 생각으로는 데이터셋을 가공하는 쪽에서 문제가 생긴 것 같아 이를 수정해야 한다.

About

Collaborative Denoising Auto Encoder with PyTorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages