For now the dataset is way too small. Need to convert Criteo Dataset to usable format and test on it.