add new stuff

Author: phinate

examples/3-blobs/3blobs1line.ipynb

@@ -0,0 +1,31 @@
+import jax.numpy as jnp
+from relax import hist_kde
+
+
+def yield_maker(data_gen, bandwidth=None, SCALE=1, syst=True):
+    def yields_from_data(angle, anchr=jnp.array([0.0, 0.0])):
+
+        s1, b1, b2, b3 = data_gen()
+
+        # direc = jnp.array([jnp.cos(angle),jnp.sin(angle)])
+        normal = jnp.array([jnp.cos(angle + jnp.pi / 2), jnp.sin(angle + jnp.pi / 2)])
+
+        bins = jnp.linspace(-100, 100, 3)
+        hb1 = hist_kde(jnp.matmul((b1 - anchr), normal), bins=bins, bandwidth=bandwidth)
+        hb2 = hist_kde(jnp.matmul((b2 - anchr), normal), bins=bins, bandwidth=bandwidth)
+        hb3 = hist_kde(jnp.matmul((b3 - anchr), normal), bins=bins, bandwidth=bandwidth)
+
+        hs1 = hist_kde(jnp.matmul((s1 - anchr), normal), bins=bins, bandwidth=bandwidth)
+
+        nb1 = hb1 / SCALE
+        nb2 = hb2 / SCALE
+        nb3 = hb3 / SCALE
+
+        ns1 = hs1 / SCALE
+
+        if not syst:
+            nb1, nb2, nb3 = nb2, nb2, nb2
+
+        return ns1, nb1, nb2, nb3
+
+    return yields_from_data

examples/3-blobs/maker.py

@@ -0,0 +1,66 @@
+import sys
+from unittest.mock import patch
+
+import pyhf
+
+jax_backend = pyhf.tensor.jax_backend(precision="64b")
+pyhf.set_backend(jax_backend)
+
+
+@patch.object(sys.modules["pyhf.tensor.common"], "default_backend", new=jax_backend)
+@patch.object(sys.modules["pyhf.pdf"], "default_backend", new=jax_backend)
+@patch("pyhf.default_backend", new=jax_backend)
+@patch.object(
+    sys.modules["pyhf.interpolators.code0"], "default_backend", new=jax_backend
+)
+@patch.object(
+    sys.modules["pyhf.interpolators.code1"], "default_backend", new=jax_backend
+)
+@patch.object(
+    sys.modules["pyhf.interpolators.code2"], "default_backend", new=jax_backend
+)
+@patch.object(
+    sys.modules["pyhf.interpolators.code4"], "default_backend", new=jax_backend
+)
+@patch.object(
+    sys.modules["pyhf.interpolators.code4p"], "default_backend", new=jax_backend
+)
+@patch.object(
+    sys.modules["pyhf.modifiers.shapefactor"], "default_backend", new=jax_backend
+)
+@patch.object(
+    sys.modules["pyhf.modifiers.shapesys"], "default_backend", new=jax_backend
+)
+@patch.object(
+    sys.modules["pyhf.modifiers.staterror"], "default_backend", new=jax_backend
+)
+@patch.object(sys.modules["pyhf.constraints"], "default_backend", new=jax_backend)
+def simplemodel2(s, b_up, b_nom, b_dn):
+    spec = {
+        "channels": [
+            {
+                "name": "singlechannel",
+                "samples": [
+                    {
+                        "name": "signal",
+                        "data": s,
+                        "modifiers": [
+                            {"name": "mu", "type": "normfactor", "data": None}
+                        ],
+                    },
+                    {
+                        "name": "background",
+                        "data": b_nom,
+                        "modifiers": [
+                            {
+                                "name": "uncorr_bkguncrt",
+                                "type": "histosys",
+                                "data": {"hi_data": b_up, "lo_data": b_dn},
+                            }
+                        ],
+                    },
+                ],
+            }
+        ]
+    }
+    return pyhf.Model(spec)

examples/3-blobs/model.py

@@ -27,13 +27,13 @@ def to_bounded(param, bounds):
 def to_inf_vec(param, bounds):
     bounds = jnp.asarray(bounds)
     a, b = bounds[:, 0], bounds[:, 1]
-    x = (2.0 * param - a) / (b - a) - 1.0
+    x = 2.0 * (param - a) / (b - a) - 1.0
     return jnp.arcsin(x)
 
 
 # [-inf, inf] <- [a,b]
 def to_inf(param, bounds):
     a, b = bounds
     # print(f"a,b: {a,b}")
-    x = (2.0 * param - a) / (b - a) - 1.0
+    x = 2.0 * (param - a) / (b - a) - 1.0
     return jnp.arcsin(x)