rime.models.zero_shot.item_knn.ItemKNN

class rime.models.zero_shot.item_knn.ItemKNN(item_df, batch_size=100, item_pop_power=1, item_pop_pseudo=0.01, model_name='bert-base-uncased', pooling=None, temperature=None, gamma=0.5)

Bases: object

softplus(x’y / temperature + log_p_y) for numerical stability

__init__(item_df, batch_size=100, item_pop_power=1, item_pop_pseudo=0.01, model_name='bert-base-uncased', pooling=None, temperature=None, gamma=0.5)

Methods

__init__(item_df[, batch_size, ...])
transform(D)