rime.dataset

Functions

prepare_minimal_dataset()	minimal dataset to enable main workflow in unit tests
prepare_simple_pattern()	Transformer(D.item_df, max_epochs=100).fit(D.auto_regressive) # flaky RNN(D.item_df, max_epochs=50).fit(D.auto_regressive) # stable
prepare_synthetic_data(split_fn_name[, ...])	prepare synthetic data for end-to-end unit tests

Modules

rime.dataset.base
rime.dataset.prepare_ml_1m_data([data_path, ...])
rime.dataset.prepare_netflix_data([...])
rime.dataset.prepare_yoochoose_data([...])