torchctr.datasets package

Submodules

torchctr.datasets.avazu module

class torchctr.datasets.avazu.Avazu[source]

Bases: torchctr.datasets.base.BaseDataset

build_data()[source]

torchctr.datasets.base module

class torchctr.datasets.base.BaseDataset[source]

Bases: torch.utils.data.dataset.Dataset

load_data()[source]
preprocess_x(non_categorical=[], non_categorical_func=None)[source]
preprocess_y(y_func=None)[source]

torchctr.datasets.criteo module

class torchctr.datasets.criteo.Criteo[source]

Bases: torchctr.datasets.base.BaseDataset

load_data()[source]

torchctr.datasets.movielens module

class torchctr.datasets.movielens.MovieLens[source]

Bases: torchctr.datasets.base.BaseDataset

build_data()[source]

torchctr.datasets.titanic module

class torchctr.datasets.titanic.Titanic[source]

Bases: torchctr.datasets.base.BaseDataset

build_data()[source]

Module contents

class torchctr.datasets.MovieLens[source]

Bases: torchctr.datasets.base.BaseDataset

build_data()[source]
class torchctr.datasets.Titanic[source]

Bases: torchctr.datasets.base.BaseDataset

build_data()[source]
class torchctr.datasets.Avazu[source]

Bases: torchctr.datasets.base.BaseDataset

build_data()[source]