ENH v6 model

des-science · May 24, 2024 · 0b7b33c · 0b7b33c
1 parent 1e1c0be
commit 0b7b33c
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/montara/make_input_desgal.py b/montara/make_input_desgal.py
@@ -373,7 +373,17 @@ def _ratio_mag_v45(mag, *coeffs):
     return 1.0 / (1.0 + np.exp(-poly))
 
 
-ratio_mag = _ratio_mag_v45
+def _ratio_mag_v6(mag, *coeffs):
+    if len(coeffs) == 0:
+        coeffs = [-4.81181605e+01,  2.49037882e-01, -3.96914217e-04,  2.16742025e-07]
+    x = mag
+    poly = np.zeros_like(x)
+    for i, c in enumerate(coeffs):
+        poly += c * x**(2*i)
+    return 1.0 / (1.0 + np.exp(-poly))
+
+
+ratio_mag = _ratio_mag_v6
 
 
 def _get_cosmos_renorm(cosmos):