KGCN

Reference:

Hongwei Wang et al. “Knowledge graph convolution networks for recommender systems.” in WWW 2019.

Reference code:

https://github.com/hwwang55/KGCN

class recbole.model.knowledge_aware_recommender.kgcn.KGCN(config, dataset)[source]

Bases: KnowledgeRecommender

KGCN is a knowledge-based recommendation model that captures inter-item relatedness effectively by mining their associated attributes on the KG. To automatically discover both high-order structure information and semantic information of the KG, we treat KG as an undirected graph and sample from the neighbors for each entity in the KG as their receptive field, then combine neighborhood information with bias when calculating the representation of a given entity.

aggregate(user_embeddings, entities, relations)[source]

For each item, aggregate the entity representation and its neighborhood representation into a single vector.

Parameters:
  • user_embeddings (torch.FloatTensor) – The embeddings of users, shape: [batch_size, embedding_size]

  • entities (list) – entities is a list of i-iter (i = 0, 1, …, n_iter) neighbors for the batch of items. dimensions of entities: {[batch_size, 1], [batch_size, n_neighbor], [batch_size, n_neighbor^2], …, [batch_size, n_neighbor^n_iter]}

  • relations (list) – relations is a list of i-iter (i = 0, 1, …, n_iter) corresponding relations for entities. relations have the same shape as entities.

Returns:

The embeddings of items, shape: [batch_size, embedding_size]

Return type:

item_embeddings(torch.FloatTensor)

calculate_loss(interaction)[source]

Calculate the training loss for a batch data.

Parameters:

interaction (Interaction) – Interaction class of the batch.

Returns:

Training loss, shape: []

Return type:

torch.Tensor

construct_adj(kg_graph)[source]

Get neighbors and corresponding relations for each entity in the KG.

Parameters:

kg_graph (scipy.sparse.coo_matrix) – an undirected graph

Returns:

  • adj_entity(torch.LongTensor): each line stores the sampled neighbor entities for a given entity, shape: [n_entities, neighbor_sample_size]

  • adj_relation(torch.LongTensor): each line stores the corresponding sampled neighbor relations, shape: [n_entities, neighbor_sample_size]

Return type:

tuple

forward(user, item)[source]

Defines the computation performed at every call.

Should be overridden by all subclasses.

Note

Although the recipe for forward pass needs to be defined within this function, one should call the Module instance afterwards instead of this since the former takes care of running the registered hooks while the latter silently ignores them.

full_sort_predict(interaction)[source]

full sort prediction function. Given users, calculate the scores between users and all candidate items.

Parameters:

interaction (Interaction) – Interaction class of the batch.

Returns:

Predicted scores for given users and all candidate items, shape: [n_batch_users * n_candidate_items]

Return type:

torch.Tensor

get_neighbors(items)[source]

Get neighbors and corresponding relations for each entity in items from adj_entity and adj_relation.

Parameters:

items (torch.LongTensor) – The input tensor that contains item’s id, shape: [batch_size, ]

Returns:

  • entities(list): Entities is a list of i-iter (i = 0, 1, …, n_iter) neighbors for the batch of items. dimensions of entities: {[batch_size, 1], [batch_size, n_neighbor], [batch_size, n_neighbor^2], …, [batch_size, n_neighbor^n_iter]}

  • relations(list): Relations is a list of i-iter (i = 0, 1, …, n_iter) corresponding relations for entities. Relations have the same shape as entities.

Return type:

tuple

input_type = 2
mix_neighbor_vectors(neighbor_vectors, neighbor_relations, user_embeddings)[source]

Mix neighbor vectors on user-specific graph.

Parameters:
  • neighbor_vectors (torch.FloatTensor) – The embeddings of neighbor entities(items), shape: [batch_size, -1, neighbor_sample_size, embedding_size]

  • neighbor_relations (torch.FloatTensor) – The embeddings of neighbor relations, shape: [batch_size, -1, neighbor_sample_size, embedding_size]

  • user_embeddings (torch.FloatTensor) – The embeddings of users, shape: [batch_size, embedding_size]

Returns:

The neighbors aggregated embeddings, shape: [batch_size, -1, embedding_size]

Return type:

neighbors_aggregated(torch.FloatTensor)

predict(interaction)[source]

Predict the scores between users and items.

Parameters:

interaction (Interaction) – Interaction class of the batch.

Returns:

Predicted scores for given users and items, shape: [batch_size]

Return type:

torch.Tensor

training: bool