Add leverages to default

stat-ml · Jun 2, 2020 · b8adac8 · b8adac8
1 parent a9583bf
commit b8adac8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alpaca/uncertainty_estimator/masks.py b/alpaca/uncertainty_estimator/masks.py
@@ -8,7 +8,7 @@
 from dppy.finite_dpps import FiniteDPP
 
 
-DEFAULT_MASKS = ['mc_dropout', 'ht_decorrelating', 'ht_dpp', 'ht_k_dpp']
+DEFAULT_MASKS = ['mc_dropout', 'ht_decorrelating', 'ht_dpp', 'ht_k_dpp', 'ht_leverages']
 
 
 # It's better to use this function to get the mask then call them directly