diff --git a/tutorials/movielens_tutorial.py b/tutorials/movielens_tutorial.py index af4e189..fb64496 100644 --- a/tutorials/movielens_tutorial.py +++ b/tutorials/movielens_tutorial.py @@ -27,10 +27,9 @@ import tensorflow as tf import pandas as pd from scipy.stats import rankdata from sklearn.model_selection import train_test_split - from tensorflow_privacy.privacy.optimizers import dp_optimizer - -from tensorflow_privacy.privacy.analysis.gdp_accountant import * +from tensorflow_privacy.privacy.analysis.gdp_accountant import compute_eps_poisson +from tensorflow_privacy.privacy.analysis.gdp_accountant import compute_mu_poisson #### FLAGS FLAGS = flags.FLAGS