37 lines
1.1 KiB
Go
37 lines
1.1 KiB
Go
// CookieJar - A contestant's algorithm toolbox
|
|
// Copyright (c) 2014 Peter Szilagyi. All rights reserved.
|
|
//
|
|
// CookieJar is dual licensed: use of this source code is governed by a BSD
|
|
// license that can be found in the LICENSE file. Alternatively, the CookieJar
|
|
// toolbox may be used in accordance with the terms and conditions contained
|
|
// in a signed written agreement between you and the author(s).
|
|
|
|
package curves
|
|
|
|
import (
|
|
"math"
|
|
|
|
"gopkg.in/karalabe/cookiejar.v2/ai/utility"
|
|
)
|
|
|
|
// Sigmoid curve builder, Defined as y = 1 / (slope*e) ^ 10(infl-x).
|
|
type Logistic struct {
|
|
Infl float64 // Point of inflection of the rate of change
|
|
Slope float64 // Multiplier changing the slope of the curve
|
|
Inc bool // Switches between increasing or decreasing curve
|
|
}
|
|
|
|
// Creates the curve mapping function.
|
|
func (l Logistic) Make() utility.Curve {
|
|
infl, slope := l.Infl, l.Slope
|
|
if l.Inc {
|
|
return func(x float64) float64 {
|
|
return 1 / (1 + math.Pow(slope*math.E, 10*(infl-x)))
|
|
}
|
|
} else {
|
|
return func(x float64) float64 {
|
|
return 1 - 1/(1+math.Pow(slope*math.E, 10*(infl-x)))
|
|
}
|
|
}
|
|
}
|