Skip to content

Commit b3f740d

Browse files
Fix tanh component (#1303)
* Fix tanh component * remove unused beta --------- Co-authored-by: Will Dean <[email protected]> Co-authored-by: Will Dean <[email protected]>
1 parent 3a98c8a commit b3f740d

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

pymc_marketing/mmm/components/saturation.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -270,14 +270,13 @@ class TanhSaturation(SaturationTransformation):
270270

271271
lookup_name = "tanh"
272272

273-
def function(self, x, b, c, beta):
273+
def function(self, x, b, c):
274274
"""Tanh saturation function."""
275-
return beta * tanh_saturation(x, b, c)
275+
return tanh_saturation(x, b, c)
276276

277277
default_priors = {
278278
"b": Prior("HalfNormal", sigma=1),
279279
"c": Prior("HalfNormal", sigma=1),
280-
"beta": Prior("HalfNormal", sigma=1),
281280
}
282281

283282

0 commit comments

Comments
 (0)