Functions — github.com/criteo-research/CausE

CodeHub MCPcopy Create free account

hub / github.com/criteo-research/CausE / functions

Functions27 in github.com/criteo-research/CausE

⨍ Functions 27 ◇ Types & classes 3

↓ 2 callersMethod__init__

src/models.py:142

↓ 2 callersFunctiongenerate_bootstrap_batch

Generate the IDs for the bootstap

src/utils.py:48

↓ 1 callersMethodadd_average_predictor

Add the average predictors to the graph

src/models.py:111

↓ 1 callersMethodadd_optimizer

Add the required optimiser to the graph

src/models.py:93

↓ 1 callersMethodadd_summaries

Add the required summaries to the graph

src/models.py:123

↓ 1 callersMethodbuild_graph

Build the main tensorflow graph with embedding layers

src/models.py:41

↓ 1 callersFunctioncompute_empircal_cr

Compute the cr from the empirical data

src/utils.py:56

↓ 1 callersMethodcreate_control_embeddings

Create the control embeddings

src/models.py:67

↓ 1 callersMethodcreate_counter_factual_loss

Create the counter factual loss to add to main loss

src/models.py:72

↓ 1 callersMethodcreate_losses

Create the losses

src/models.py:77

↓ 1 callersFunctioncreate_movielens_userproduct_matrix

Create the matrix from the movielens dataset

src/Data/dataset_loading.py:53

↓ 1 callersMethodcreate_placeholders

Create the placeholders to be used

src/models.py:30

↓ 1 callersFunctionload_movielens10M

Load the movielens dataset from disk

src/Data/dataset_loading.py:30

src/models.py:7

src/models.py:171

Functioncompute_2i_regularization_id

Compute the ID for the regularization for the 2i approach

src/utils.py:66

Functioncompute_bootstraps

Compute the bootstraps for the 0 indexed model

src/utils.py:140

Functioncompute_bootstraps_2i

Compute the bootstraps for the 2i model

src/utils.py:92

Functioncompute_treatment_or_control

Compute if product is in treatment or control

src/utils.py:79

Methodcreate_control_embeddings

Create the control embeddings

src/models.py:146

Methodcreate_control_embeddings

Create the control embeddings

src/models.py:175

Methodcreate_counter_factual_loss

Create the counter factual loss to add to main loss

src/models.py:153

Methodcreate_counter_factual_loss

Create the counter factual loss to add to main loss

src/models.py:182

Functionload_movielens

Load the movielens 100K dataset from disk

src/Data/dataset_loading.py:9

Functionload_netflix_dataset

(dataset_location)

src/Data/dataset_loading.py:81

Functionload_test_dataset

Load the test and validation datasets

src/utils.py:29

Functionload_train_dataset

Load the training data using TF Dataset API

src/utils.py:10