Personalizing Recommendations

In the last tutorial step, for each song, we computed the songs most frequently played next. It is now time to use this data to make actual recommendations. We simply look at the last song each user has listened to and recommend the most popular song played after it.

Specifically, the songs-table has information about the top next songs in its column info:top_next_songs. We have the latest song a user has played in the users-table in column info:track_plays. We now have to join these to get our result.

The next sections walk through the pipeline line-by-line and describe the custom functions as they appear. The entire file is available at the end of the page.

We first describe a helper function which retrieves the song id of the most popular song given the list of top next songs. getMostPopularSong takes as its input a KijiSlice of AvroRecords from the “info:top_next_songs” column of the songs table.

The “info:top_next_songs” column contains a TopSongs Avro record. You can look in the KijiMusic.avdl file for the definitions of these Avro records:

  /** A count of the number of times a song has been played. */
  record SongCount {
    string song_id;
    long count;
  }

  /** Container record for the top songs and their number of plays. */
  record TopSongs {
     array<SongCount> top_songs;
  }

In KijiExpress, Avro records are read out as a generic AvroRecord. Fields in a record can be accessed with record("fieldname"); if the result of that is a list or map, that can be accessed by index or key, respectively. If the result of that is a primitive, we need to declare what type of primitive we expect it to be with, for example, asString, if it is a String.

For the getMostPopularSongs function, we get the latest version of the slice using .getFirstValue. We know we the contents of the “info:top_next_songs” column will be a TopSongs record with a field called “top_songs” that is a list of the next songs. We access that field with songRecord("top_songs"). We know those top next songs are records with a field “song_id” that is the ID of that song, and it is a string, so we retrieve it with mostPopularSong("song_id").asString. The entire function looks like this:

  // This method retrieves the most popular song (at index 0) in the TopNextSongs record.
  def getMostPopularSong(songs: KijiSlice[AvroRecord]): String = {
    val songRecord = songs.getFirstValue
    val topSongs = songRecord("top_songs")
    val mostPopularSong = topSongs(0)
    return mostPopularSong("song_id").asString
  }

Apply the above function on a Scalding Pipe

Next, we then construct a Scalding RichPipe does the following:

  • Reads the column info:top_next_songs from the songs table as 'topNextSongs.
  • Uses the first component of the EntityId as the songId.
  • Applies the above described getMostPopularSong to all the songs.
  • Outputs tuples of (songId, nextSong)
  val recommendedSong = KijiInput(args("songs-table"))("info:top_next_songs" -> 'topNextSongs)
    .map('entityId -> 'songId) { eId: EntityId => eId(0) }
    .map('topNextSongs -> 'nextSong) { getMostPopularSong}
    .project('songId, 'nextSong)

Putting it all together

Finally we create a flow that does the following:

  • Reads the column “info:track_plays” from the users table, which contains the listening history.
  • Retrieves the song most recently played by a user into the field 'lastTrackPlayed.
  • Generate the recommendation by joining the two pipes, namely, one containing a (user, song id) representing the latest song played, and another containing (song id, next song) representing the most popular next song.
  KijiInput(args("users-table"))("info:track_plays" -> 'trackPlays)
      .map('trackPlays -> 'lastTrackPlayed) {
           slice: KijiSlice[String] => slice.getFirstValue()}
      .joinWithSmaller('lastTrackPlayed -> 'songId, recommendedSong)
      .write(KijiOutput(args("users-table"))('nextSong -> "info:next_song_rec"))

Running the Example

  • To run the SongRecommender job:
express job ${MUSIC_EXPRESS_HOME}/lib/kiji-express-music-0.6.0.jar \
    org.kiji.express.music.SongRecommender --songs-table ${KIJI}/songs \
    --users-table ${KIJI}/users

Alternative: Running as a script

Alternately, you can also run the job as a script as follows:

express script --libjars "${MUSIC_EXPRESS_HOME}/lib/* \
    ${MUSIC_EXPRESS_HOME}/scripts/SongRecommender.express --hdfs

Verify Output

You can verify the output by scanning the users-table.

kiji scan ${KIJI}/users --max-rows=2

You should see something like:

Scanning kiji table: kiji://localhost:2181/kiji_express_music/users/
entity-id=['user-41'] [1325762580000] info:track_plays
                                 song-41
entity-id=['user-41'] [1367023207962] info:next_song_rec
                                 song-41

Shut down the cluster

That’s the end of the Express tutorial.

  • Now is a good time to shut down the BentoBox cluster:
    bento stop

Top Next Songs Job Content

Here’s the entire SongRecommender job:

SongPlayCounter.express