dataset

Dataset - all data container.

Classes

BaseFeaturesSchema(*, names)

Features schema.

Dataset(user_id_map, item_id_map, interactions)

Container class for all data for a recommendation model.

DatasetSchema(*, n_interactions, users, items)

Dataset schema.

DenseFeaturesSchema(*, names[, kind])

Dense features schema.

EntitySchema(*, n_hot, id_map[, features])

Entity schema.

IdMapSchema(*, size, dtype)

IdMap schema.

SparseFeaturesSchema(*, names[, kind])

Sparse features schema.