poprox_recommender.model.nrms.user_encoder

Classes

UserEncoder(hidden_size, num_attention_heads)

class poprox_recommender.model.nrms.user_encoder.UserEncoder(hidden_size, num_attention_heads)

Bases: Module

forward(article_vectors)

Parameters:

news_vectors – batch_size, num_clicked_news_a_user, word_embedding_dim

Returns:

(shape) batch_size, word_embedding_dim