Add aliases for powerbald and powerentropy

baal/active/heuristics/aliases.py

@@ -0,0 +1,17 @@
+"""Some heuristics are simply combinations of others."""
+from typing import Union, Callable
+
+from baal.active.heuristics import BALD, Entropy
+from baal.active.heuristics.stochastics import PowerSampling
+
+
+def PowerBALD(query_size: int, temperature: float = 1.0, reduction: Union[str, Callable] = "none"):
+    return PowerSampling(BALD(reduction=reduction), query_size=query_size, temperature=temperature)
+
+
+def PowerEntropy(
+    query_size: int, temperature: float = 1.0, reduction: Union[str, Callable] = "none"
+):
+    return PowerSampling(
+        Entropy(reduction=reduction), query_size=query_size, temperature=temperature
+    )

docs/user_guide/heuristics.md

@@ -33,11 +33,9 @@ This approach is on-par with methods like BADGE or BatchBALD, while being hundre
     To use PowerBALD in your code, it is as simple as:
 
     ```
-    from baal.active.heuristics import BALD
-    from baal.active.heuristics.stochastics import PowerSampling
+    from baal.active.heuristics.aliases import PowerBALD
 
-
-    heuristic = PowerSampling(BALD(), query_size=QUERY_SIZE, temperature=1.0)
+    heuristic = PowerBALD(query_size=QUERY_SIZE, temperature=1.0)
     ```
 
 ### BALD