diff options
Diffstat (limited to 'lib/sigmoid.c')
| -rw-r--r-- | lib/sigmoid.c | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/lib/sigmoid.c b/lib/sigmoid.c index 05ea126..33fd58b 100644 --- a/lib/sigmoid.c +++ b/lib/sigmoid.c @@ -44,10 +44,11 @@ int ts_sigmoid( { TS_PROLOGUE( ts_sigmoid_p ); if ( !c ) return -1; - fp_t m = ((ts_sigmoid_params_t *)c)->m; - fp_t L = ((ts_sigmoid_params_t *)c)->L; - fp_t b = ((ts_sigmoid_params_t *)c)->b; + fp_t m = ((ts_sigmoid_params_t const *)c)->m; + fp_t L = ((ts_sigmoid_params_t const *)c)->L; + fp_t b = ((ts_sigmoid_params_t const *)c)->b; + if (L == 0 || isnan(L) || isinf(L) || isnan(m) || isinf(m)) return -1; m = -( 4.0 * m ) / L; b = b - ( 0.5 * L ); |