src.models.components.click_predictor package§

Submodules§

src.models.components.click_predictor.DNN module§

class src.models.components.click_predictor.DNN.DNNClickPredictor(input_size, hidden_size=None)§

Bases: Module

forward(candidate_news_vector, user_vector)§
Parameters:

  • candidate_news_vector – batch_size, X

  • user_vector – batch_size, X

Returns:

batch_size

Return type:

(shape)

training: bool§

src.models.components.click_predictor.dot_product module§

class src.models.components.click_predictor.dot_product.DotProductClickPredictor§

Bases: Module

forward(candidate_news_vector, user_vector)§
Parameters:

  • candidate_news_vector – batch_size, candidate_size, X

  • user_vector – batch_size, X

Returns:

batch_size

Return type:

(shape)

training: bool§

Module contents§