Z

WSDM_Spotify

public

1 stars

0 forks

0 issues

Commits

List of commits on branch master.

Verified

94135feb909e1be75f973e891850ec8434b6eb9d

Update README.md

ZZhaoyueCheng committed 3 years ago

Unverified

4a0e551f0229dfefacbb07d7f65f46a897db582d

Create WSDM_Challenge.pdf

ZZhaoyueCheng committed 3 years ago

Verified

f15f8c9fbae433f1508dbe5d68c75a353c0417b4

Update README.md

ZZhaoyueCheng committed 4 years ago

Verified

57aa8affee792fa9ff71c0c8a485f3b8e4ca0a05

Update README.md

ZZhaoyueCheng committed 6 years ago

Unverified

b56cb686310378283fadc94a4520355c02a89de9

refactor

ZZhaoyueCheng committed 6 years ago

Unverified

a7b061cc2844eba6c81f5cc0760ef67167b6ba3b

cleaned up the preprocess folder and scripts

ZZhaoyueCheng committed 6 years ago

README

The README file for this repository.

ACM WSDM Spotify Skip Prediction Challenge 4th place solution: Utilizing Attention Mechanism for Sequential Skip Prediction [paper]

Team Name: learner6
Team Members: Zhaoyue Cheng, Maksims Volkovs

Environment and Requirements

The model is implemented and tested on the following Environment:

Intel(R) Xeon(R) CPU E5-2620 v4 @ 2.10GHz
256GB RAM
Nvidia Titan Xp

With Requirements:

Python 3.6
Numpy
Pandas
Pytorch 0.4.1 with Cuda 9.0

Preprocessing and Executing

All the required code are uploaded, instructions to be updated soon
The network we used is in folder models/rnn_model, we code up multiple different models but only be able to test one of them due to time constraints, the one we are using for the leaderboard submission is RNNModelAtt1SeperateEncoder in file rnn_model which corresponds to the Network Architecture image shown below

Network Architecture