rime.dataset.base.DatasetBase

class rime.dataset.base.DatasetBase(user_df: DataFrame, item_df: DataFrame, event_df: Optional[DataFrame] = None)

Bases: object

base class contains the global set of users, items, and events for auto-regressive training

__init__(user_df: DataFrame, item_df: DataFrame, event_df: Optional[DataFrame] = None) → None

Methods

__init__(user_df, item_df[, event_df])

Attributes

event_df
user_df
item_df