rime.dataset.base.Dataset

class rime.dataset.base.Dataset(user_df: DataFrame, item_df: DataFrame, event_df: Optional[DataFrame] = None, test_requests: Optional[DataFrame] = None, item_in_test: Optional[DataFrame] = None, horizon: float = inf, target_csr: Optional[spmatrix] = None, exclude_train: Union[bool, list] = True, add_prior_to_form_reranking_task: float = 0, prior_score: Optional[spmatrix] = None, _skip_init: InitVar = False, test_update_history: bool = True, sample_with_prior: Optional[float] = None)[source]

Bases: DatasetBase

Dataset with holdout targets; allows multiple requests by the same user

__init__(user_df: DataFrame, item_df: DataFrame, event_df: Optional[DataFrame] = None, test_requests: Optional[DataFrame] = None, item_in_test: Optional[DataFrame] = None, horizon: float = inf, target_csr: Optional[spmatrix] = None, exclude_train: Union[bool, list] = True, add_prior_to_form_reranking_task: float = 0, prior_score: Optional[spmatrix] = None, _skip_init: InitVar = False, test_update_history: bool = True, sample_with_prior: Optional[float] = None) → None

Methods

`__init__`(user_df, item_df[, event_df, ...])
`get_stats`()
`print_stats`([verbose])
`reindex`(index, axis)
`reindex_unbiased`([min_user_len, ...])
`sample`(, axis, *kw)

Attributes

`add_prior_to_form_reranking_task`
`auto_regressive`
`event_df`
`exclude_train`
`horizon`
`item_in_test`
`item_ppl_baseline`
`prior_score`
`sample_with_prior`
`shape`
`target_csr`
`test_requests`
`test_update_history`
`user_in_test`	alias with simple index
`user_ppl_baseline`

property user_in_test: alias with simple index